Skip to content

Commit f179583

Browse files
gate the reranker test
1 parent 6b558ec commit f179583

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(reranker):
7979
with pytest.raises(Exception):
8080
reranker.rank("valid query", "not a list") # Invalid type for documents
8181

82-
with pytest.raises(Exception):
83-
reranker.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)