Skip to content

Commit 1d978ce

Browse files
FIX: conserve n_samples attribute
1 parent 40731a9 commit 1d978ce

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

mapie/classification.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1104,6 +1104,7 @@ def fit(
11041104
sample_weight,
11051105
groups,
11061106
n_samples) = self._check_fit_parameter(X, y, sample_weight, groups)
1107+
self.n_samples_ = n_samples
11071108

11081109
if self.method == "raps":
11091110
(X, y_enc, y, n_samples, sample_weight, groups) = self._split_data(

0 commit comments

Comments
 (0)