Skip to content

Commit 7d744f3

Browse files
committed
Fix for CI
1 parent 935b449 commit 7d744f3

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

optuna_integration/sklearn/sklearn.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -280,7 +280,10 @@ def _cross_validate_with_pruning(
280280
if is_classifier(estimator):
281281
partial_fit_params = self.fit_params.copy()
282282
y = self.y.values if isinstance(self.y, pd.Series) else self.y
283-
classes = np.unique(y)
283+
if y is not None:
284+
classes = np.unique(y)
285+
else:
286+
classes = None
284287

285288
partial_fit_params.setdefault("classes", classes)
286289

0 commit comments

Comments
 (0)