Run all tests and fix nmslib (#73)

scikit-learn-contrib · Jan 13, 2025 · 0342c23 · 0342c23
1 parent eaae17c
commit 0342c23
Show file tree

Hide file tree

Showing 9 changed files with 3 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -40,7 +40,7 @@ pynndescent = [
     "numba>=0.52",
 ]
 nmslib = [
-    "nmslib>=2.1.1,<3.0.0 ; python_version < '3.11'",
+    "nmslib-metabrainz>=2.1.1,<3.0.0",
 ]
 annlibs = [
     "sklearn-ann[annoy,faiss,pynndescent,nmslib]",

diff --git a/src/sklearn_ann/cluster/tests/__init__.py b/src/sklearn_ann/cluster/tests/__init__.py
diff --git a/src/sklearn_ann/kneighbors/nmslib.py b/src/sklearn_ann/kneighbors/nmslib.py
@@ -2,7 +2,7 @@
 import numpy as np
 from scipy.sparse import csr_matrix
 from sklearn.base import BaseEstimator, TransformerMixin
-from sklearn.utils import Tags, TransformerTags
+from sklearn.utils import Tags, TargetTags, TransformerTags
 from sklearn.utils.validation import validate_data
 
 from ..utils import TransformerChecksMixin, check_metric
@@ -67,5 +67,6 @@ def transform(self, X):
     def __sklearn_tags__(self) -> Tags:
         return Tags(
             estimator_type="transformer",
+            target_tags=TargetTags(required=False),
             transformer_tags=TransformerTags(preserves_dtype=[np.float32]),
         )
diff --git a/src/sklearn_ann/cluster/tests/test_common.py → tests/test_cluster.py b/src/sklearn_ann/cluster/tests/test_common.py → tests/test_cluster.py
diff --git a/tests/conftest.py → tests/test_kneighbors/conftest.py b/tests/conftest.py → tests/test_kneighbors/conftest.py
diff --git a/tests/test_annoy.py → tests/test_kneighbors/test_annoy.py b/tests/test_annoy.py → tests/test_kneighbors/test_annoy.py
diff --git a/tests/test_common.py → tests/test_kneighbors/test_common.py b/tests/test_common.py → tests/test_kneighbors/test_common.py
diff --git a/tests/test_faiss.py → tests/test_kneighbors/test_faiss.py b/tests/test_faiss.py → tests/test_kneighbors/test_faiss.py
diff --git a/tests/test_nmslib.py → tests/test_kneighbors/test_nmslib.py b/tests/test_nmslib.py → tests/test_kneighbors/test_nmslib.py