Skip to content

Commit 516fd33

Browse files
check reranker type first before test
1 parent 1cbe7a6 commit 516fd33

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

tests/integration/test_rerankers.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -79,10 +79,11 @@ def test_bad_input_cohere(cohereReranker):
7979
with pytest.raises(Exception):
8080
cohereReranker.rank("valid query", "not a list") # Invalid type for documents
8181

82-
with pytest.raises(Exception):
83-
cohereReranker.rank(
84-
"valid query", [{"field": "valid document"}], rank_by=["invalid_field"]
85-
) # Invalid rank_by field
82+
if isinstance(reranker, CohereReranker):
83+
with pytest.raises(Exception):
84+
reranker.rank(
85+
"valid query", [{"field": "valid document"}], rank_by=["invalid_field"]
86+
) # Invalid rank_by field
8687

8788

8889
def test_rank_documents_cross_encoder(hfCrossEncoderReranker):

0 commit comments

Comments
 (0)