fix cohere reranker test

Author: tylerhutcherson

tests/integration/test_rerankers.py

@@ -33,6 +33,10 @@ def reranker(request, skip_reranker):
         return request.param(model="rerank-lite-1")
 
 
+@pytest.fixture
+def cohereReranker():
+    return CohereReranker()
+
 @pytest.fixture
 def hfCrossEncoderReranker():
     return HFCrossEncoderReranker()
@@ -79,11 +83,10 @@ def test_bad_input_cohere(cohereReranker):
     with pytest.raises(Exception):
         cohereReranker.rank("valid query", "not a list")  # Invalid type for documents
 
-    if isinstance(reranker, CohereReranker):
-        with pytest.raises(Exception):
-            reranker.rank(
-                "valid query", [{"field": "valid document"}], rank_by=["invalid_field"]
-            )  # Invalid rank_by field
+    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):