fix: method assignment

AnswerDotAI · Mar 18, 2024 · 555aaf3 · 555aaf3
1 parent 884ed64
commit 555aaf3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ragatouille/models/index.py b/ragatouille/models/index.py
@@ -130,8 +130,8 @@ class HNSWModelIndex(ModelIndex):
 class PLAIDModelIndex(ModelIndex):
     _DEFAULT_INDEX_BSIZE = 32
     index_type = "PLAID"
-    faiss_kmeans = deepcopy(CollectionIndexer._train_kmeans)
-    pytorch_kmeans = torch_kmeans._train_kmeans
+    faiss_kmeans = staticmethod(deepcopy(CollectionIndexer._train_kmeans))
+    pytorch_kmeans = staticmethod(torch_kmeans._train_kmeans)
 
     def __init__(self, config: ColBERTConfig) -> None:
         super().__init__(config)