fix nmslib.py typos

mldrugdiscovery · Jun 22, 2020 · 4795f3e · 4795f3e
1 parent 6db359e
commit 4795f3e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/ann_benchmarks/.DS_Store b/ann_benchmarks/.DS_Store
diff --git a/ann_benchmarks/algorithms/nmslib.py b/ann_benchmarks/algorithms/nmslib.py
@@ -66,6 +66,8 @@ def fit(self, X):
                 X_trans = NmslibReuseIndex.matrToStrArray(csr_matrix(X))
             else:
                 X_trans = NmslibReuseIndex.intMatrToStrArray(X)
+            self._index = nmslib.init(space=self._nmslib_metric, method=self._method_name, data_type=nmslib.DataType.OBJECT_AS_STRING)
+            self._index.addDataPointBatch(X_trans)
         else:
             self._index = nmslib.init(space=self._nmslib_metric, method=self._method_name)
             self._index.addDataPointBatch(X)