format

deepset-ai · Dec 22, 2023 · 2e690e4 · 2e690e4
1 parent 72570ed
commit 2e690e4
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/integrations/pinecone/src/pinecone_haystack/document_store.py b/integrations/pinecone/src/pinecone_haystack/document_store.py
@@ -164,7 +164,9 @@ def write_documents(self, documents: List[Document], policy: DuplicatePolicy = D
     def filter_documents(self, filters: Optional[Dict[str, Any]] = None) -> List[Document]:
         if not filters:
             # in this case, we try to return all documents but Pinecone has some limits
-            documents = self._embedding_retrieval(query_embedding=self._dummy_vector, namespace=self.namespace, top_k=TOP_K_LIMIT)
+            documents = self._embedding_retrieval(
+                query_embedding=self._dummy_vector, namespace=self.namespace, top_k=TOP_K_LIMIT
+            )
             for doc in documents:
                 doc.score = None
 
@@ -190,7 +192,7 @@ def _embedding_retrieval(
         self,
         query_embedding: List[float],
         *,
-        namespace: Optional[str]=None,
+        namespace: Optional[str] = None,
         filters: Optional[Dict[str, Any]] = None,  # noqa: ARG002 (filters to be implemented)
         top_k: int = 10,
     ) -> List[Document]: