[CI] Bump mteb version to mteb[bm25s]>=2, <3 for pooling model unit tests (#34961)

Signed-off-by: yewentao256 <zhyanwentao@126.com>
2026-02-21 23:23:24 -05:00
parent d403c1da1c
commit d24bdd7c4b
4 changed files with 6 additions and 3 deletions
--- a/tests/models/language/pooling_mteb_test/mteb_score_utils.py
+++ b/tests/models/language/pooling_mteb_test/mteb_score_utils.py
@@ -191,6 +191,9 @@ def run_mteb_rerank(cross_encoder: mteb.CrossEncoderProtocol, tasks, languages):
        mteb_tasks: list[mteb.abstasks.AbsTaskRetrieval] = mteb.get_tasks(
            tasks=tasks, languages=languages, eval_splits=eval_splits
        )
+        for task in mteb_tasks:
+            if not task.data_loaded:
+                task.load_data()

        mteb.evaluate(
            bm25s,