From 476a664e711cf9d0bbc0264a4f5cd4bac2f52155 Mon Sep 17 00:00:00 2001 From: Karem Abdul-Samad Date: Wed, 1 Jan 2025 21:34:03 -0500 Subject: [PATCH] fix: Torch MPS backend failing test (#20709) --- keras/src/utils/backend_utils_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/keras/src/utils/backend_utils_test.py b/keras/src/utils/backend_utils_test.py index 6255f0d7bd73..248831046017 100644 --- 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)