Skip to content

Commit 5979dcb

Browse files
committed
ENH: check group lenght in check fit params
1 parent d7e88c5 commit 5979dcb

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

mapie/mondrian.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -145,7 +145,6 @@ def fit(
145145
"""
146146

147147
X, y, groups = self._check_fit_parameters(X, y, groups)
148-
self._check_group_length(X, groups)
149148
self.unique_groups = np.unique(groups)
150149
self.mapie_estimators = {}
151150

@@ -424,5 +423,6 @@ def _check_fit_parameters(
424423
groups = cast(NDArray, np.array(groups))
425424

426425
self._check_groups_fit(X, groups)
426+
self._check_group_length(X, groups)
427427

428428
return X, y, groups

0 commit comments

Comments
 (0)