fix: Torch MPS backend failing test (#20709)

kas2020-commits · web-flow · commit 476a664e711c · 2025-01-01T18:34:03.000-08:00
diff --git a/keras/src/utils/backend_utils_test.py b/keras/src/utils/backend_utils_test.py
@@ -15,7 +15,7 @@ class BackendUtilsTest(testing.TestCase):
     )
     def test_dynamic_backend(self, name):
         dynamic_backend = backend_utils.DynamicBackend()
-        x = np.random.uniform(size=[1, 2, 3])
+        x = np.random.uniform(size=[1, 2, 3]).astype("float32")
 
         if name == "numpy":
             dynamic_backend.set_backend(name)

Original file line number	Diff line number	Diff line change
`@@ -15,7 +15,7 @@ class BackendUtilsTest(testing.TestCase):`
`15`	`15`	`)`
`16`	`16`	`def test_dynamic_backend(self, name):`
`17`	`17`	`dynamic_backend = backend_utils.DynamicBackend()`
`18`		`- x = np.random.uniform(size=[1, 2, 3])`
	`18`	`+ x = np.random.uniform(size=[1, 2, 3]).astype("float32")`
`19`	`19`
`20`	`20`	`if name == "numpy":`
`21`	`21`	`dynamic_backend.set_backend(name)`