pep8

sdpython · Mar 18, 2019 · b20d37c · b20d37c
1 parent bc9a9b6
commit b20d37c
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 8 deletions.
diff --git a/_unittests/ut_mlmodel/test_decision_tree_experiment.py b/_unittests/ut_mlmodel/test_decision_tree_experiment.py
@@ -29,14 +29,14 @@
 
 
 if compare_module_version(sklearn.__version__, "0.21") >= 0:
-    from src.mlinsights.mlmodel.piecewise_tree_regression_criterion import SimpleRegressorCriterion  # pylint: disable=E0611
-    from src.mlinsights.mlmodel.piecewise_tree_regression_criterion import _test_criterion_init  # pylint: disable=E0611
-    from src.mlinsights.mlmodel.piecewise_tree_regression_criterion import _test_criterion_node_impurity  # pylint: disable=E0611
-    from src.mlinsights.mlmodel.piecewise_tree_regression_criterion import _test_criterion_node_impurity_children  # pylint: disable=E0611
-    from src.mlinsights.mlmodel.piecewise_tree_regression_criterion import _test_criterion_update  # pylint: disable=E0611
-    from src.mlinsights.mlmodel.piecewise_tree_regression_criterion import _test_criterion_node_value  # pylint: disable=E0611
-    from src.mlinsights.mlmodel.piecewise_tree_regression_criterion import _test_criterion_proxy_impurity_improvement  # pylint: disable=E0611
-    from src.mlinsights.mlmodel.piecewise_tree_regression_criterion import _test_criterion_impurity_improvement  # pylint: disable=E0611
+    from src.mlinsights.mlmodel.piecewise_tree_regression_criterion import SimpleRegressorCriterion
+    from src.mlinsights.mlmodel.piecewise_tree_regression_criterion import _test_criterion_init
+    from src.mlinsights.mlmodel.piecewise_tree_regression_criterion import _test_criterion_node_impurity
+    from src.mlinsights.mlmodel.piecewise_tree_regression_criterion import _test_criterion_node_impurity_children
+    from src.mlinsights.mlmodel.piecewise_tree_regression_criterion import _test_criterion_update
+    from src.mlinsights.mlmodel.piecewise_tree_regression_criterion import _test_criterion_node_value
+    from src.mlinsights.mlmodel.piecewise_tree_regression_criterion import _test_criterion_proxy_impurity_improvement
+    from src.mlinsights.mlmodel.piecewise_tree_regression_criterion import _test_criterion_impurity_improvement
 
 
 class TestDecisionTreeExperiment(ExtTestCase):

diff --git a/_unittests/ut_module/test_code_style.py b/_unittests/ut_module/test_code_style.py
@@ -56,6 +56,8 @@ def test_style_test(self):
                          "Instance of 'tuple' has no",
                          "[E402] module level import",
                          "R1720",
+                         "E0611: No name '_test_criterion_",
+                         "E0611: No name 'SimpleRegressorCriterion'",
                          ])