Pylint, update tests

deepset-ai · Mar 5, 2024 · 3cb0cf2 · 3cb0cf2
1 parent c5226e2
commit 3cb0cf2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/...s/mongodb_atlas/src/haystack_integrations/document_stores/mongodb_atlas/document_store.py b/...s/mongodb_atlas/src/haystack_integrations/document_stores/mongodb_atlas/document_store.py
@@ -51,6 +51,7 @@ class MongoDBAtlasDocumentStore:
     print(store.count_documents())
     ```
     """
+
     def __init__(
         self,
         *,

diff --git a/integrations/mongodb_atlas/tests/test_retriever.py b/integrations/mongodb_atlas/tests/test_retriever.py
@@ -102,6 +102,6 @@ def test_run(self):
         retriever = MongoDBAtlasEmbeddingRetriever(document_store=mock_store)
         res = retriever.run(query_embedding=[0.3, 0.5])
 
-        mock_store.embedding_retrieval.assert_called_once_with(query_embedding_np=[0.3, 0.5], filters={}, top_k=10)
+        mock_store.embedding_retrieval.assert_called_once_with(query_embedding=[0.3, 0.5], filters={}, top_k=10)
 
         assert res == {"documents": [doc]}
-Original file line number
+Diff line change
@@ Expand Up / @@ -51,6 +51,7 @@ class MongoDBAtlasDocumentStore: @@
         print(store.count_documents())
         ```
         """
         def __init__(
             self,
             *,
@@ Expand Down @@