Skip to content

Commit be87487

Browse files
superbobrychiruu12
authored andcommitted
TrackingTest no longer assigns None to variables (keras-team#21106)
JAX will soon fail when `jnp.array` is called with None, so this test will be broken under newer JAX versions if kept as is.
1 parent 53b9866 commit be87487

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

keras/src/utils/tracking_test.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ def test_untracking_in_tracked_list(self):
2020
v2 = backend.Variable(2.0)
2121
lst = tracking.TrackedList([], tracker)
2222
lst.append(v1)
23-
lst.append(None)
23+
lst.append(float("nan"))
2424
lst.append(v2)
2525
lst.append(0)
2626

@@ -38,7 +38,7 @@ def test_untracking_in_tracked_list(self):
3838

3939
lst2 = tracking.TrackedList([], tracker)
4040
lst2.append(v1)
41-
lst2.append(None)
41+
lst2.append(float("nan"))
4242
lst2.append(v2)
4343
lst2.append(0)
4444

0 commit comments

Comments
 (0)