fix: Torch MPS backend failing test (#20709)

keras-team · Jan 2, 2025 · 476a664 · 476a664
1 parent 59d3025
commit 476a664
Showing 1 changed file with 1 addition and 1 deletion.
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)