Skip to content

Commit 7e0a0cf

Browse files
author
Frankie Robertson
committed
Add workaround for PyNNDescentTransformer pickling problems
1 parent 87d08b9 commit 7e0a0cf

File tree

1 file changed

+12
-1
lines changed

1 file changed

+12
-1
lines changed

sklearn_ann/kneighbors/pynndescent.py

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,15 @@
1-
from pynndescent import PyNNDescentTransformer
1+
from pynndescent import PyNNDescentTransformer as PyNNDescentTransformerBase
2+
3+
4+
def no_op():
5+
pass
6+
7+
8+
class PyNNDescentTransformer(PyNNDescentTransformerBase):
9+
def fit(self, X, compress_index=True):
10+
super().fit(X, compress_index=compress_index)
11+
self.index_.compress_index = no_op
12+
return self
213

314

415
__all__ = ["PyNNDescentTransformer"]

0 commit comments

Comments
 (0)