chore: format & minor fixes

AnswerDotAI · Feb 5, 2024 · 18c080e · 18c080e
1 parent a81d344
commit 18c080e
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "RAGatouille"
-version = "0.0.6b4"
+version = "0.0.6b5"
 description = "Library to facilitate the use of state-of-the-art retrieval models in common RAG contexts."
 authors = ["Benjamin Clavie <[email protected]>"]
 readme = "README.md"

diff --git a/ragatouille/__init__.py b/ragatouille/__init__.py
@@ -1,4 +1,4 @@
-__version__ = "0.0.6b4"
+__version__ = "0.0.6b5"
 from .RAGPretrainedModel import RAGPretrainedModel
 from .RAGTrainer import RAGTrainer
 

diff --git a/ragatouille/models/colbert.py b/ragatouille/models/colbert.py
@@ -438,7 +438,7 @@ def search(
         k: int = 10,
         force_fast: bool = False,
         zero_index_ranks: bool = False,
-        doc_ids: Optional[List[str]] = None
+        doc_ids: Optional[List[str]] = None,
     ):
         if self.searcher is None or (
             index_name is not None and self.index_name != index_name