correct mypy

scikit-learn-contrib · Nov 29, 2024 · 8334a86 · 8334a86
1 parent 6c4f1c5
commit 8334a86
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/mapie/regression/regression.py b/mapie/regression/regression.py
@@ -579,6 +579,8 @@ def fit_estimator(
             **self._fit_params
         )
 
+        return self
+
     def conformlize(
         self,
         X: ArrayLike,