Skip flaky TF test

keras-team · Dec 4, 2024 · 44e0723 · 44e0723
1 parent 5354cf9
commit 44e0723
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/keras/src/trainers/data_adapters/py_dataset_adapter_test.py b/keras/src/trainers/data_adapters/py_dataset_adapter_test.py
@@ -93,6 +93,7 @@ def __getitem__(self, index):
         raise ValueError("Expected exception")
 
 
+@pytest.mark.skipif(testing.tensorflow_uses_gpu(), reason="Flaky on GPU")
 class PyDatasetAdapterTest(testing.TestCase):
     @parameterized.named_parameters(
         named_product(