From fedf29596653aee3911c8ffe6b72487ca6870ca5 Mon Sep 17 00:00:00 2001
From: michaelfeil <me@michaelfeil.eu>
Date: Fri, 7 Jun 2024 20:54:47 -0700
Subject: [PATCH] fix: pytest faster

---
 libs/infinity_emb/tests/unit_test/test_engine.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/libs/infinity_emb/tests/unit_test/test_engine.py b/libs/infinity_emb/tests/unit_test/test_engine.py
index 8ebb46b0..ee727d31 100644
--- a/libs/infinity_emb/tests/unit_test/test_engine.py
+++ b/libs/infinity_emb/tests/unit_test/test_engine.py
@@ -199,7 +199,7 @@ async def test_async_api_torch_usage():
     [
         EmbeddingDtype.float32,
         EmbeddingDtype.int8,
-        EmbeddingDtype.uint8,
+        # EmbeddingDtype.uint8,
         EmbeddingDtype.ubinary,
     ],
 )
@@ -210,6 +210,7 @@ async def test_async_api_torch_embedding_quant(embedding_dtype: EmbeddingDtype):
         device = "cpu"
     engine = AsyncEmbeddingEngine.from_args(
         EngineArgs(
+            model_name_or_path="TaylorAI/gte-tiny",
             engine=InferenceEngine.torch,
             device=Device[device],
             lengths_via_tokenize=True,