conert to set

mlfoundations · Oct 22, 2023 · 003d357 · 003d357
1 parent 20a2cb3
commit 003d357
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_generation.py b/tests/test_generation.py
@@ -13,7 +13,7 @@
     torch._C._jit_set_profiling_executor(True)
     torch._C._jit_set_profiling_mode(False)
 
-models_to_test = open_clip.list_generative_models().difference(
+models_to_test = set(open_clip.list_generative_models()).difference(
     {"coca_roberta-ViT-B-32", "coca_base"}
 )