fix unit tests for python 3.6 and sklearn latest (#70)

scikit-learn-contrib · Jun 4, 2021 · a06bc1a · a06bc1a
1 parent 8b3715b
commit a06bc1a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/skltemplate/_template.py b/skltemplate/_template.py
@@ -27,7 +27,7 @@ class TemplateEstimator(BaseEstimator):
     >>> y = np.zeros((100, ))
     >>> estimator = TemplateEstimator()
     >>> estimator.fit(X, y)
-    TemplateEstimator(demo_param='demo_param')
+    TemplateEstimator()
     """
     def __init__(self, demo_param='demo_param'):
         self.demo_param = demo_param

diff --git a/skltemplate/tests/test_common.py b/skltemplate/tests/test_common.py
@@ -8,7 +8,8 @@
 
 
 @pytest.mark.parametrize(
-    "Estimator", [TemplateEstimator, TemplateTransformer, TemplateClassifier]
+    "estimator",
+    [TemplateEstimator(), TemplateTransformer(), TemplateClassifier()]
 )
-def test_all_estimators(Estimator):
-    return check_estimator(Estimator)
+def test_all_estimators(estimator):
+    return check_estimator(estimator)