File tree Expand file tree Collapse file tree 1 file changed +8
-0
lines changed
keras/src/layers/preprocessing Expand file tree Collapse file tree 1 file changed +8
-0
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments