Fix NMSLib transformer

scikit-learn-contrib · Jan 13, 2025 · cf156f6 · cf156f6
1 parent 9fef4ac
commit cf156f6
Showing 1 changed file with 2 additions and 1 deletion.
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]),
         )