Follow up with kwargs propagation due to change in parent class

run-llama · Jul 2, 2024 · c26d6f8 · c26d6f8
1 parent 722cb67
commit c26d6f8
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/...s/indices/llama-index-indices-managed-colbert/llama_index/indices/managed/colbert/base.py b/...s/indices/llama-index-indices-managed-colbert/llama_index/indices/managed/colbert/base.py
@@ -105,7 +105,9 @@ def as_retriever(self, **kwargs: Any) -> BaseRetriever:
     def ref_doc_info(self) -> Dict[str, RefDocInfo]:
         raise NotImplementedError("ColbertStoreIndex does not support ref_doc_info.")
 
-    def _build_index_from_nodes(self, nodes: Sequence[BaseNode]) -> IndexDict:
+    def _build_index_from_nodes(
+        self, nodes: Sequence[BaseNode], **kwargs: Any
+    ) -> IndexDict:
         """Generate a PLAID index from the ColBERT checkpoint via its hugging face
         model_name.
         """