[Test] Skip tokenizer tests if transformers is not in workspace

ghstack-source-id: dfb9792ca1680ef4bc00aa3e80eafe456274362d Pull Request resolved: #2744
pytorch · Feb 3, 2025 · ffc0d1e · ffc0d1e
1 parent 8ba5fbd
commit ffc0d1e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/test/test_transforms.py b/test/test_transforms.py
@@ -191,6 +191,7 @@
 TIMEOUT = 100.0
 
 _has_gymnasium = importlib.util.find_spec("gymnasium") is not None
+_has_transformers = importlib.util.find_spec("transformers") is not None
 
 
 class TransformBase:
@@ -2532,6 +2533,9 @@ def test_transform_inverse(self):
             )
 
 
+@pytest.mark.skipif(
+    not _has_transformers, reason="transformers needed to test tokenizers"
+)
 class TestTokenizer(TransformBase):
     @pytest.mark.parametrize("datatype", ["str", "NonTensorStack"])
     def test_transform_no_env(self, datatype):