check reranker type first before test

redis · Oct 1, 2024 · 8b1b520 · 8b1b520
1 parent ee68921
commit 8b1b520
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/tests/integration/test_rerankers.py b/tests/integration/test_rerankers.py
@@ -79,10 +79,11 @@ def test_bad_input(reranker):
     with pytest.raises(Exception):
         reranker.rank("valid query", "not a list")  # Invalid type for documents
 
-    with pytest.raises(Exception):
-        reranker.rank(
-            "valid query", [{"field": "valid document"}], rank_by=["invalid_field"]
-        )  # Invalid rank_by field
+    if isinstance(reranker, CohereReranker):
+        with pytest.raises(Exception):
+            reranker.rank(
+                "valid query", [{"field": "valid document"}], rank_by=["invalid_field"]
+            )  # Invalid rank_by field
 
 
 def test_rank_documents_cross_encoder(hfCrossEncoderReranker):