Make faiss dependency optional.

easeml · Sep 30, 2024 · 7228cbb · 7228cbb
1 parent 0291437
commit 7228cbb
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/experiments/datascope/experiments/pipelines/models.py b/experiments/datascope/experiments/pipelines/models.py
@@ -1,4 +1,4 @@
-import faiss
+# import faiss
 import numpy as np
 import tempfile
 import torch
@@ -100,6 +100,8 @@ def __init__(self, n_neighbors=5):
         self.label_encoder = LabelEncoder()
 
     def fit(self, X, y):
+        import faiss
+
         self.X = X.astype(np.float32)
         self.y = self.label_encoder.fit_transform(y)
         d = self.X.shape[1]