From 20a19fe2ad88e8df3af8061f169c69476189ed1a Mon Sep 17 00:00:00 2001
From: Vincent Moens <vmoens@meta.com>
Date: Mon, 3 Feb 2025 17:16:10 +0000
Subject: [PATCH] [Test] Skip tokenizer tests if transformers is not in
 workspace

ghstack-source-id: b92facfd14cba62511e7888567c94d3986419ab5
Pull Request resolved: https://github.com/pytorch/rl/pull/2744
---
 test/test_transforms.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/test/test_transforms.py b/test/test_transforms.py
index f57bc58221d..038f284ed19 100644
--- 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):