Skip to content

Commit 7878bb2

Browse files
committed
iter
1 parent a33b9f8 commit 7878bb2

File tree

2 files changed

+3
-6
lines changed

2 files changed

+3
-6
lines changed

imblearn/ensemble/tests/test_easy_ensemble.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
"""Test the module easy ensemble."""
2+
23
# Authors: Guillaume Lemaitre <[email protected]>
34
# Christos Aridas
45
# License: MIT
@@ -211,9 +212,5 @@ def test_easy_ensemble_classifier_grid_search():
211212
"n_estimators": [1, 2],
212213
"estimator__n_estimators": [3, 4],
213214
}
214-
grid_search = GridSearchCV(
215-
EasyEnsembleClassifier(estimator=AdaBoostClassifier()),
216-
parameters,
217-
cv=5,
218-
)
215+
grid_search = GridSearchCV(EasyEnsembleClassifier(), parameters, cv=5)
219216
grid_search.fit(X, y)

imblearn/pipeline.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1059,7 +1059,7 @@ def _fit_resample_one(sampler, X, y, message_clsname="", message=None, params=No
10591059
return X_res, y_res, sampler
10601060

10611061

1062-
def _transform_one(transformer, X, y, weight, params):
1062+
def _transform_one(transformer, X, y, weight, params=None):
10631063
"""Call transform and apply weight to output.
10641064
10651065
Parameters

0 commit comments

Comments
 (0)