snip long prompt in tests

finegrain-ai · Mar 5, 2024 · a0be545 · a0be545
1 parent defbb9e
commit a0be545
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/foundationals/clip/test_text_encoder.py b/tests/foundationals/clip/test_text_encoder.py
@@ -24,7 +24,7 @@
 PROMPTS = [
     "",  # empty
     "a cute cat",  # padded
-    long_prompt,  # truncated
+    "<long prompt>",  # see above, truncated
     "64k",  # FG-362 - encoded as 3 tokens
 ]
 
@@ -67,7 +67,7 @@ def test_basics(ref_tokenizer: transformers.CLIPTokenizer, our_encoder: CLIPText
 
 @pytest.fixture(params=PROMPTS)
 def prompt(request: pytest.FixtureRequest):
-    return request.param
+    return long_prompt if request.param == "<long prompt>" else request.param
 
 
 def test_encoder(