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

ghstack-source-id: b92facfd14cba62511e7888567c94d3986419ab5 Pull Request resolved: #2744
pytorch · Feb 3, 2025 · 20a19fe · 20a19fe · github-actions · Feb 3, 2025
1 parent 10f015e
commit 20a19fe
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):