Skip to content

Commit b614fc2

Browse files
[CI] Stabilize pooling rerank equivalence test
Signed-off-by: Taneem Ibrahim <taneem.ibrahim@gmail.com>
1 parent 26d725c commit b614fc2

1 file changed

Lines changed: 4 additions & 2 deletions

File tree

tests/entrypoints/pooling/scoring/test_cross_encoder_online_vision.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -473,7 +473,7 @@ async def test_rerank_api_instruction_field(
473473
async def test_rerank_api_instruction_field_matches_chat_template_kwargs(
474474
server: tuple[RemoteOpenAIServer, str],
475475
):
476-
remote_server, _ = server
476+
remote_server, backend = server
477477

478478
doc_list = [
479479
document,
@@ -514,4 +514,6 @@ async def test_rerank_api_instruction_field_matches_chat_template_kwargs(
514514
kwargs_scores = [
515515
r.relevance_score for r in sorted(kwargs_rerank.results, key=lambda x: x.index)
516516
]
517-
assert field_scores == pytest.approx(kwargs_scores)
517+
assert field_scores == pytest.approx(
518+
kwargs_scores, rel=get_tol(backend), abs=get_abs_tol(backend)
519+
)

0 commit comments

Comments
 (0)