Skip to content

Commit 7fc4176

Browse files
committed
pep8
1 parent 61bfe8a commit 7fc4176

File tree

2 files changed

+17
-9
lines changed

2 files changed

+17
-9
lines changed

sklearn/tests/test_common.py

Lines changed: 15 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,8 @@ def test_all_estimators():
5656

5757
for name, Estimator in estimators:
5858
# 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)
6061

6162

6263
def test_non_meta_estimators():
@@ -115,7 +116,8 @@ def test_class_weight_balanced_linear_classifiers():
115116
issubclass(clazz, LinearClassifierMixin))]
116117

117118
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
119121

120122

121123
@ignore_warnings
@@ -197,8 +199,9 @@ def test_non_transformer_estimators_n_iter():
197199
else:
198200
# Multitask models related to ENet cannot handle
199201
# 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)
202205

203206

204207
def test_transformer_n_iter():
@@ -219,9 +222,11 @@ def test_transformer_n_iter():
219222
if isinstance(estimator, ProjectedGradientNMF):
220223
# The ProjectedGradientNMF class is deprecated
221224
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
223227
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
225230

226231

227232
def test_get_params_invariance():
@@ -236,6 +241,8 @@ def test_get_params_invariance():
236241
# If class is deprecated, ignore deprecated warnings
237242
if hasattr(Estimator.__init__, "deprecated_original"):
238243
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
240246
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

sklearn/utils/estimator_checks.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,8 @@
7676

7777

7878
def _set_test_name(function, name):
79-
function.description = "sklearn.tests.test_common.{0}({1})".format(function.__name__, name)
79+
function.description = ("sklearn.tests.test_common.{0}({1})".format(
80+
function.__name__, name))
8081
return function
8182

8283

0 commit comments

Comments
 (0)