Skip to content

Commit cf156f6

Browse files
committed
Fix NMSLib transformer
1 parent 9fef4ac commit cf156f6

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

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
)

0 commit comments

Comments
 (0)