From 4d3b2d9cf3223adcfb8de7d90981b80d1c6e5d12 Mon Sep 17 00:00:00 2001 From: hh-space-invader Date: Thu, 27 Feb 2025 10:14:31 +0200 Subject: [PATCH] fix: Fix CI --- tests/test_text_cross_encoder.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/tests/test_text_cross_encoder.py b/tests/test_text_cross_encoder.py index d2fbce73..7c6ac00a 100644 --- a/tests/test_text_cross_encoder.py +++ b/tests/test_text_cross_encoder.py @@ -16,7 +16,7 @@ } -def test_rerank(model_name: str) -> None: +def test_rerank() -> None: is_ci = os.getenv("CI") is_manual = os.getenv("GITHUB_EVENT_NAME") == "workflow_dispatch" @@ -28,9 +28,8 @@ def test_rerank(model_name: str) -> None: ) for model_desc in models_to_test: - if ( - not is_ci and model_desc.size_in_GB > 1 - ) or model_desc.model not in CANONICAL_SCORE_VALUES: + model_name = model_desc.model + if (not is_ci and model_desc.size_in_GB > 1) or model_name not in CANONICAL_SCORE_VALUES: continue model = TextCrossEncoder(model_name=model_name)