@@ -56,7 +56,8 @@ def test_all_estimators():
56
56
57
57
for name , Estimator in estimators :
58
58
# some can just not be sensibly default constructed
59
- yield _set_test_name (check_parameters_default_constructible , name ), name , Estimator
59
+ yield (_set_test_name (check_parameters_default_constructible , name ),
60
+ name , Estimator )
60
61
61
62
62
63
def test_non_meta_estimators ():
@@ -115,7 +116,8 @@ def test_class_weight_balanced_linear_classifiers():
115
116
issubclass (clazz , LinearClassifierMixin ))]
116
117
117
118
for name , Classifier in linear_classifiers :
118
- yield _set_test_name (check_class_weight_balanced_linear_classifier , name ), name , Classifier
119
+ yield _set_test_name (check_class_weight_balanced_linear_classifier ,
120
+ name ), name , Classifier
119
121
120
122
121
123
@ignore_warnings
@@ -197,8 +199,9 @@ def test_non_transformer_estimators_n_iter():
197
199
else :
198
200
# Multitask models related to ENet cannot handle
199
201
# if y is mono-output.
200
- yield (_set_test_name (check_non_transformer_estimators_n_iter , name ),
201
- name , estimator , 'Multi' in name )
202
+ yield (_set_test_name (
203
+ check_non_transformer_estimators_n_iter , name ),
204
+ name , estimator , 'Multi' in name )
202
205
203
206
204
207
def test_transformer_n_iter ():
@@ -219,9 +222,11 @@ def test_transformer_n_iter():
219
222
if isinstance (estimator , ProjectedGradientNMF ):
220
223
# The ProjectedGradientNMF class is deprecated
221
224
with ignore_warnings ():
222
- yield _set_test_name (check_transformer_n_iter , name ), name , estimator
225
+ yield _set_test_name (
226
+ check_transformer_n_iter , name ), name , estimator
223
227
else :
224
- yield _set_test_name (check_transformer_n_iter , name ), name , estimator
228
+ yield _set_test_name (
229
+ check_transformer_n_iter , name ), name , estimator
225
230
226
231
227
232
def test_get_params_invariance ():
@@ -236,6 +241,8 @@ def test_get_params_invariance():
236
241
# If class is deprecated, ignore deprecated warnings
237
242
if hasattr (Estimator .__init__ , "deprecated_original" ):
238
243
with ignore_warnings ():
239
- yield _set_test_name (check_get_params_invariance , name ), name , Estimator
244
+ yield _set_test_name (
245
+ check_get_params_invariance , name ), name , Estimator
240
246
else :
241
- yield _set_test_name (check_get_params_invariance , name ), name , Estimator
247
+ yield _set_test_name (
248
+ check_get_params_invariance , name ), name , Estimator
0 commit comments