TST remove double check of sciki-learn estimator (scikit-learn-contri…

…b#339)
souravsingh · Aug 31, 2017 · 7344bea · 7344bea
1 parent 894de60
commit 7344bea
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/imblearn/utils/estimator_checks.py b/imblearn/utils/estimator_checks.py
@@ -19,8 +19,7 @@
 
 from sklearn.datasets import make_classification
 from sklearn.cluster import KMeans
-from sklearn.utils.estimator_checks import _yield_all_checks \
-    as sklearn_yield_all_checks, check_estimator \
+from sklearn.utils.estimator_checks import check_estimator \
     as sklearn_check_estimator, check_parameters_default_constructible
 from sklearn.exceptions import NotFittedError
 from sklearn.utils.testing import assert_allclose
@@ -49,8 +48,6 @@ def _yield_sampler_checks(name, Estimator):
 
 
 def _yield_all_checks(name, Estimator):
-    # make the checks from scikit-learn
-    sklearn_yield_all_checks(name, Estimator)
     # trigger our checks if this is a SamplerMixin
     if issubclass(Estimator, SamplerMixin):
         for check in _yield_sampler_checks(name, Estimator):