improve: Prevent stop iteration error caused by next

hh-space-invader · hh-space-invader · commit 0b673a304731 · 2025-02-27T10:21:18.000+02:00
diff --git a/tests/test_image_onnx_embeddings.py b/tests/test_image_onnx_embeddings.py
@@ -34,7 +34,7 @@ def test_embedding() -> None:
     all_models = ImageEmbedding._list_supported_models()
 
     models_to_test = (
-        [next(model for model in all_models if model.model in CANONICAL_VECTOR_VALUES)]
+        [model for model in all_models if model.model in CANONICAL_VECTOR_VALUES][:1]
         if not is_manual
         else all_models
     )
diff --git a/tests/test_late_interaction_embeddings.py b/tests/test_late_interaction_embeddings.py
@@ -177,7 +177,7 @@ def test_single_embedding():
 
     all_models = LateInteractionTextEmbedding._list_supported_models()
     models_to_test = (
-        [next(model for model in all_models if model.model in CANONICAL_COLUMN_VALUES)]
+        [model for model in all_models if model.model in CANONICAL_COLUMN_VALUES][:1]
         if not is_manual
         else all_models
     )
@@ -206,7 +206,7 @@ def test_single_embedding_query():
 
     all_models = LateInteractionTextEmbedding._list_supported_models()
     models_to_test = (
-        [next(model for model in all_models if model.model in CANONICAL_QUERY_VALUES)]
+        [model for model in all_models if model.model in CANONICAL_QUERY_VALUES][:1]
         if not is_manual
         else all_models
     )
diff --git a/tests/test_sparse_embeddings.py b/tests/test_sparse_embeddings.py
@@ -70,7 +70,7 @@ def test_single_embedding() -> None:
 
     all_models = SparseTextEmbedding._list_supported_models()
     models_to_test = (
-        [next(model for model in all_models if model.model in CANONICAL_COLUMN_VALUES)]
+        [model for model in all_models if model.model in CANONICAL_COLUMN_VALUES][:1]
         if not is_manual
         else all_models
     )
diff --git a/tests/test_text_cross_encoder.py b/tests/test_text_cross_encoder.py
@@ -22,7 +22,7 @@ def test_rerank() -> None:
 
     all_models = TextCrossEncoder._list_supported_models()
     models_to_test = (
-        [next(model for model in all_models if model.model in CANONICAL_SCORE_VALUES)]
+        [model for model in all_models if model.model in CANONICAL_SCORE_VALUES][:1]
         if not is_manual
         else all_models
     )
diff --git a/tests/test_text_multitask_embeddings.py b/tests/test_text_multitask_embeddings.py
@@ -93,7 +93,7 @@ def test_single_embedding():
 
     all_models = TextEmbedding._list_supported_models()
     models_to_test = (
-        [next(model for model in all_models if model.model in CANONICAL_VECTOR_VALUES)]
+        [model for model in all_models if model.model in CANONICAL_VECTOR_VALUES][:1]
         if not is_manual
         else all_models
     )
@@ -133,7 +133,7 @@ def test_single_embedding_query():
 
     all_models = TextEmbedding._list_supported_models()
     models_to_test = (
-        [next(model for model in all_models if model.model in CANONICAL_VECTOR_VALUES)]
+        [model for model in all_models if model.model in CANONICAL_VECTOR_VALUES][:1]
         if not is_manual
         else all_models
     )
@@ -172,7 +172,7 @@ def test_single_embedding_passage():
 
     all_models = TextEmbedding._list_supported_models()
     models_to_test = (
-        [next(model for model in all_models if model.model in CANONICAL_VECTOR_VALUES)]
+        [model for model in all_models if model.model in CANONICAL_VECTOR_VALUES][:1]
         if not is_manual
         else all_models
     )
diff --git a/tests/test_text_onnx_embeddings.py b/tests/test_text_onnx_embeddings.py
@@ -80,7 +80,7 @@ def test_embedding() -> None:
     all_models = TextEmbedding._list_supported_models()
 
     models_to_test = (
-        [next(model for model in all_models if model.model in CANONICAL_VECTOR_VALUES)]
+        [model for model in all_models if model.model in CANONICAL_VECTOR_VALUES][:1]
         if not is_manual
         else all_models
     )

Original file line number	Diff line number	Diff line change
`@@ -34,7 +34,7 @@ def test_embedding() -> None:`
`34`	`34`	`all_models = ImageEmbedding._list_supported_models()`
`35`	`35`
`36`	`36`	`models_to_test = (`
`37`		`- [next(model for model in all_models if model.model in CANONICAL_VECTOR_VALUES)]`
	`37`	`+ [model for model in all_models if model.model in CANONICAL_VECTOR_VALUES][:1]`
`38`	`38`	`if not is_manual`
`39`	`39`	`else all_models`
`40`	`40`	`)`
Original file line number	Diff line number	Diff line change
`@@ -70,7 +70,7 @@ def test_single_embedding() -> None:`
`70`	`70`
`71`	`71`	`all_models = SparseTextEmbedding._list_supported_models()`
`72`	`72`	`models_to_test = (`
`73`		`- [next(model for model in all_models if model.model in CANONICAL_COLUMN_VALUES)]`
	`73`	`+ [model for model in all_models if model.model in CANONICAL_COLUMN_VALUES][:1]`
`74`	`74`	`if not is_manual`
`75`	`75`	`else all_models`
`76`	`76`	`)`
Original file line number	Diff line number	Diff line change
`@@ -22,7 +22,7 @@ def test_rerank() -> None:`
`22`	`22`
`23`	`23`	`all_models = TextCrossEncoder._list_supported_models()`
`24`	`24`	`models_to_test = (`
`25`		`- [next(model for model in all_models if model.model in CANONICAL_SCORE_VALUES)]`
	`25`	`+ [model for model in all_models if model.model in CANONICAL_SCORE_VALUES][:1]`
`26`	`26`	`if not is_manual`
`27`	`27`	`else all_models`
`28`	`28`	`)`
Original file line number	Diff line number	Diff line change
`@@ -80,7 +80,7 @@ def test_embedding() -> None:`
`80`	`80`	`all_models = TextEmbedding._list_supported_models()`
`81`	`81`
`82`	`82`	`models_to_test = (`
`83`		`- [next(model for model in all_models if model.model in CANONICAL_VECTOR_VALUES)]`
	`83`	`+ [model for model in all_models if model.model in CANONICAL_VECTOR_VALUES][:1]`
`84`	`84`	`if not is_manual`
`85`	`85`	`else all_models`
`86`	`86`	`)`