fix: replace mocking class method with object method

qdrant · Feb 20, 2025 · 50210fc · 50210fc
1 parent 7ad4c5e
commit 50210fc
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/tests/embed_tests/test_local_inference.py b/tests/embed_tests/test_local_inference.py
@@ -1672,10 +1672,9 @@ def decorated(*args, **kwargs):
     local_client = QdrantClient(":memory:", local_inference_batch_size=inference_batch_size)
     if not local_client._FASTEMBED_INSTALLED:
         pytest.skip("FastEmbed is not installed, skipping")
-    # local_client._model_embedder.embedder.embed = mock(
-    #     local_client._model_embedder.embedder.embed, param_storage
-    # )
-    Embedder.embed = mock(Embedder.embed, param_storage)
+    local_client._model_embedder.embedder.embed = mock(
+        local_client._model_embedder.embedder.embed, param_storage
+    )
 
     if not local_client._FASTEMBED_INSTALLED:
         pytest.skip("FastEmbed is not installed, skipping")