From 671b874d8ef5c8dabb5085be81418338e1eedc36 Mon Sep 17 00:00:00 2001
From: hh-space-invader <h.hagag.ali@gmail.com>
Date: Fri, 28 Feb 2025 14:40:19 +0200
Subject: [PATCH] fix: Fix test in image model

---
 tests/test_image_onnx_embeddings.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/tests/test_image_onnx_embeddings.py b/tests/test_image_onnx_embeddings.py
index 565bcd7f..cc72fb20 100644
--- 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"
     ],
 )