We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9fef4ac commit cf156f6Copy full SHA for cf156f6
src/sklearn_ann/kneighbors/nmslib.py
@@ -2,7 +2,7 @@
2
import numpy as np
3
from scipy.sparse import csr_matrix
4
from sklearn.base import BaseEstimator, TransformerMixin
5
-from sklearn.utils import Tags, TransformerTags
+from sklearn.utils import Tags, TargetTags, TransformerTags
6
from sklearn.utils.validation import validate_data
7
8
from ..utils import TransformerChecksMixin, check_metric
@@ -67,5 +67,6 @@ def transform(self, X):
67
def __sklearn_tags__(self) -> Tags:
68
return Tags(
69
estimator_type="transformer",
70
+ target_tags=TargetTags(required=False),
71
transformer_tags=TransformerTags(preserves_dtype=[np.float32]),
72
)
0 commit comments