fix bug

scikit-learn-contrib · Dec 19, 2024 · 87441b0 · 87441b0
1 parent 34d9cdc
commit 87441b0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mapie/regression/quantile_regression.py b/mapie/regression/quantile_regression.py
@@ -538,7 +538,7 @@ def fit(
             self._fit_estimators(
                 X=X_train,
                 y=y_train,
-                sample_weight_train=sample_weight,
+                sample_weight=sample_weight,
                 **fit_params
             )