We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e310fa7 commit 2cee211Copy full SHA for 2cee211
imblearn/utils/estimator_checks.py
@@ -35,6 +35,8 @@
35
36
from imblearn.utils.testing import warns
37
38
+NOT_TESTED_SAMPLERS = ('FunctionSampler') # This is not a normal sampler
39
+
40
41
def _yield_sampler_checks(name, Estimator):
42
yield check_target_type
@@ -76,7 +78,8 @@ def check_estimator(Estimator):
76
78
sklearn_check_estimator(Estimator)
77
79
check_parameters_default_constructible(name, Estimator)
80
for check in _yield_all_checks(name, Estimator):
- check(name, Estimator)
81
+ if name not in NOT_TESTED_SAMPLERS:
82
+ check(name, Estimator)
83
84
85
def check_target_type(name, Estimator):
0 commit comments