fix: Fix test in image model

hh-space-invader · hh-space-invader · commit 671b874d8ef5 · 2025-02-28T14:40:19.000+02:00
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"
     ],
 )

Original file line number	Diff line number	Diff line change
`@@ -27,13 +27,14 @@`
`27`	`27`	`}`
`28`	`28`
`29`	`29`	`ALL_IMAGE_MODEL_DESC = ImageEmbedding._list_supported_models()`
	`30`	`+smallest_model = min(ALL_IMAGE_MODEL_DESC, key=lambda m: m.size_in_GB).model`
`30`	`31`
`31`	`32`
`32`	`33`	`@pytest.mark.parametrize(`
`33`	`34`	`"model_name",`
`34`	`35`	`[`
`35`		`- min(ALL_IMAGE_MODEL_DESC, key=lambda m: m.size_in_GB).model`
`36`		`- if CANONICAL_VECTOR_VALUES`
	`36`	`+ smallest_model`
	`37`	`+ if smallest_model in CANONICAL_VECTOR_VALUES`
`37`	`38`	`else "Qdrant/clip-ViT-B-32-vision"`
`38`	`39`	`],`
`39`	`40`	`)`