Skip failing TF test on GPU

Author: fchollet

keras/src/export/onnx_test.py

@@ -77,6 +77,9 @@ def get_model(type="sequential", input_shape=(10,), layer_list=None):
     ),
 )
 @pytest.mark.skipif(testing.jax_uses_gpu(), reason="Leads to core dumps on CI")
+@pytest.mark.skipif(
+    testing.tensorflow_uses_gpu(), reason="Leads to core dumps on CI"
+)
 class ExportONNXTest(testing.TestCase):
     @parameterized.named_parameters(
         named_product(