Merge pull request #2966 from Garve/master

Update sklearn.py
optuna · Sep 30, 2021 · 9bbe9c8 · 9bbe9c8
2 parents 59163b8 + 8656f0f
commit 9bbe9c8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/optuna/integration/sklearn.py b/optuna/integration/sklearn.py
@@ -838,7 +838,7 @@ def fit(
             fit_params_res = _check_fit_params(X, fit_params, self.sample_indices_)
 
         classifier = is_classifier(self.estimator)
-        cv = check_cv(self.cv, y_res, classifier)
+        cv = check_cv(self.cv, y_res, classifier=classifier)
 
         self.n_splits_ = cv.get_n_splits(X_res, y_res, groups=groups_res)
         self.scorer_ = check_scoring(self.estimator, scoring=self.scoring)