diff --git a/tests/testing_utils.py b/tests/testing_utils.py index a00f5f954..cf05b1ece 100644 --- 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