fix: Fix test in image model

qdrant · Feb 28, 2025 · 671b874 · 671b874
1 parent da8f392
commit 671b874
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/test_image_onnx_embeddings.py b/tests/test_image_onnx_embeddings.py
@@ -27,13 +27,14 @@
 }
 
 ALL_IMAGE_MODEL_DESC = ImageEmbedding._list_supported_models()
+smallest_model = min(ALL_IMAGE_MODEL_DESC, key=lambda m: m.size_in_GB).model
 
 
 @pytest.mark.parametrize(
     "model_name",
     [
-        min(ALL_IMAGE_MODEL_DESC, key=lambda m: m.size_in_GB).model
-        if CANONICAL_VECTOR_VALUES
+        smallest_model
+        if smallest_model in CANONICAL_VECTOR_VALUES
         else "Qdrant/clip-ViT-B-32-vision"
     ],
 )