Fix CI

keras-team · Sep 26, 2024 · cbe0bea · cbe0bea
1 parent 09a85a2
commit cbe0bea
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/keras/src/dtype_policies/dtype_policy_map_test.py b/keras/src/dtype_policies/dtype_policy_map_test.py
@@ -11,6 +11,7 @@
 from keras.src.dtype_policies.dtype_policy_map import DTypePolicyMap
 
 
+@pytest.mark.skipif(testing.jax_uses_gpu(), reason="Leads to core dumps on CI")
 class DTypePolicyMapTest(testing.TestCase):
     def setUp(self):
         super().setUp()