22
33from dataclasses import dataclass , field
44
5+ from .fixes import _dataclass_args
6+
57# Authors: The scikit-learn developers
68# SPDX-License-Identifier: BSD-3-Clause
79
810
9- @dataclass
11+ @dataclass ( ** _dataclass_args ())
1012class InputTags :
1113 """Tags for the input data.
1214
@@ -67,7 +69,7 @@ class InputTags:
6769 pairwise : bool = False
6870
6971
70- @dataclass
72+ @dataclass ( ** _dataclass_args ())
7173class TargetTags :
7274 """Tags for the target data.
7375
@@ -106,7 +108,7 @@ class TargetTags:
106108 single_output : bool = True
107109
108110
109- @dataclass
111+ @dataclass ( ** _dataclass_args ())
110112class TransformerTags :
111113 """Tags for the transformer.
112114
@@ -126,7 +128,7 @@ class TransformerTags:
126128 preserves_dtype : list [str ] = field (default_factory = lambda : ["float64" ])
127129
128130
129- @dataclass
131+ @dataclass ( ** _dataclass_args ())
130132class ClassifierTags :
131133 """Tags for the classifier.
132134
@@ -154,7 +156,7 @@ class ClassifierTags:
154156 multi_label : bool = False
155157
156158
157- @dataclass
159+ @dataclass ( ** _dataclass_args ())
158160class RegressorTags :
159161 """Tags for the regressor.
160162
@@ -176,7 +178,7 @@ class RegressorTags:
176178 multi_label : bool = False
177179
178180
179- @dataclass
181+ @dataclass ( ** _dataclass_args ())
180182class Tags :
181183 """Tags for the estimator.
182184
0 commit comments