Skip to content

Commit 0342c23

Browse files
authored
Run all tests and fix nmslib (#73)
1 parent eaae17c commit 0342c23

File tree

9 files changed

+3
-2
lines changed

9 files changed

+3
-2
lines changed

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ pynndescent = [
4040
"numba>=0.52",
4141
]
4242
nmslib = [
43-
"nmslib>=2.1.1,<3.0.0 ; python_version < '3.11'",
43+
"nmslib-metabrainz>=2.1.1,<3.0.0",
4444
]
4545
annlibs = [
4646
"sklearn-ann[annoy,faiss,pynndescent,nmslib]",

src/sklearn_ann/cluster/tests/__init__.py

Whitespace-only changes.

src/sklearn_ann/kneighbors/nmslib.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
import numpy as np
33
from scipy.sparse import csr_matrix
44
from sklearn.base import BaseEstimator, TransformerMixin
5-
from sklearn.utils import Tags, TransformerTags
5+
from sklearn.utils import Tags, TargetTags, TransformerTags
66
from sklearn.utils.validation import validate_data
77

88
from ..utils import TransformerChecksMixin, check_metric
@@ -67,5 +67,6 @@ def transform(self, X):
6767
def __sklearn_tags__(self) -> Tags:
6868
return Tags(
6969
estimator_type="transformer",
70+
target_tags=TargetTags(required=False),
7071
transformer_tags=TransformerTags(preserves_dtype=[np.float32]),
7172
)
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.

0 commit comments

Comments
 (0)