Skip to content

Commit 7dc5962

Browse files
committed
fix: CI testing failures
1 parent 569159d commit 7dc5962

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

keras/src/layers/preprocessing/text_vectorization_test.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -493,6 +493,10 @@ def test_adapt_with_steps(self):
493493
self.assertIn("bar", vocab)
494494
self.assertNotIn("unique_word", vocab)
495495

496+
@pytest.mark.skipif(
497+
backend.backend() != "tensorflow",
498+
reason="TextVectorization and Grain adapt path require TensorFlow",
499+
)
496500
def test_adapt_with_grain_dataset(self):
497501
pytest.importorskip("grain")
498502
import grain as grain_module
@@ -521,6 +525,10 @@ def __getitem__(self, index):
521525
self.assertIn("bar", vocab)
522526
self.assertIn("baz", vocab)
523527

528+
@pytest.mark.skipif(
529+
backend.backend() != "tensorflow",
530+
reason="TextVectorization and Grain adapt path require TensorFlow",
531+
)
524532
def test_adapt_with_grain_dataset_and_steps(self):
525533
pytest.importorskip("grain")
526534
import grain as grain_module

0 commit comments

Comments
 (0)