Covered all pylint errors and warnings.

deepsense-ai · Sep 20, 2024 · 4f2fbfe · 4f2fbfe
1 parent 717052d
commit 4f2fbfe
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 168 deletions.
diff --git a/packages/ragbits-document-search/src/ragbits/document_search/vector_store/chromadb_store.py b/packages/ragbits-document-search/src/ragbits/document_search/vector_store/chromadb_store.py
@@ -141,7 +141,7 @@ async def retrieve(self, vector: List[float], k: int = 5) -> List[VectorDBEntry]
 
         Args:
             vector (List[float]): The vector to query.
-            k (int, default=5): The number of entries to retrieve.
+            k (int): The number of entries to retrieve.
 
         Returns:
             List[VectorDBEntry]: The retrieved entries.

diff --git a/packages/ragbits-document-search/tests/test_chroma_db.py b/packages/ragbits-document-search/tests/test_chroma_db.py