fix

Signed-off-by: Brian Dellabetta <[email protected]>
vllm-project · Feb 14, 2025 · 23499d9 · 23499d9
1 parent f37cfeb
commit 23499d9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/testing_utils.py b/tests/testing_utils.py
@@ -231,7 +231,7 @@ def tokenize(sample):
                 add_special_tokens=False,
             )
 
-    elif ds_name in ("flickr30k",):
+    elif ds_name == "flickr30k":
         def preprocess(sample):
             return sample