From 1b066019a8eaef102e74c395ac09422d3f0b0760 Mon Sep 17 00:00:00 2001 From: Alexey Vatolin Date: Wed, 25 Dec 2024 15:44:33 +0100 Subject: [PATCH] Speed up test_save_predictions (#1631) --- tests/test_cli.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_cli.py b/tests/test_cli.py index 842fa9368d..4f13bebbc6 100644 --- a/tests/test_cli.py +++ b/tests/test_cli.py @@ -189,7 +189,7 @@ def test_create_meta_from_existing(existing_readme_name: str, gold_readme_name: def test_save_predictions(): - command = f"{sys.executable} -m mteb run -m all-MiniLM-L6-v2 -t NFCorpus --output_folder tests/results --save_predictions" + command = f"{sys.executable} -m mteb run -m average_word_embeddings_komninos -t NFCorpus --output_folder tests/results --save_predictions" result = subprocess.run(command, shell=True, capture_output=True, text=True) assert result.returncode == 0, "Command failed" test_folder = Path(__file__).parent