fix: more speedup on windows

michaelfeil · Jun 8, 2024 · f5226aa · f5226aa
1 parent fedf295
commit f5226aa
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/libs/infinity_emb/tests/unit_test/inference/test_models.py b/libs/infinity_emb/tests/unit_test/inference/test_models.py
@@ -52,7 +52,7 @@ def _pretrained_model_score(
 @pytest.mark.parametrize(
     "model,score,compute_type",
     [
-        ("sentence-transformers/bert-base-nli-mean-tokens", 76.40, "int8"),
+        ("sentence-transformers/bert-base-nli-mean-tokens", 76.37, "int8"),
         ("sentence-transformers/bert-base-nli-mean-tokens", 76.46, None),
         ("sentence-transformers/all-MiniLM-L6-v2", 81.03, None),
         ("sentence-transformers/all-MiniLM-L6-v2", 81.03, "default"),

diff --git a/libs/infinity_emb/tests/unit_test/test_engine.py b/libs/infinity_emb/tests/unit_test/test_engine.py
@@ -194,6 +194,7 @@ async def test_async_api_torch_usage():
 
 
 @pytest.mark.anyio
+@pytest.mark.skipif(sys.platform != "linux", reason="only run these on Linux")
 @pytest.mark.parametrize(
     "embedding_dtype",
     [