ADD: re-add old import path for EnsembleRegressor

Author: vincentblot28

mapie/estimator/estimator.py

@@ -0,0 +1,12 @@
+from sklearn.utils import deprecated
+
+from mapie.estimator.regressor import EnsembleRegressor as NewEnsembleRegressor
+
+
+@deprecated(
+    "WARNING: Deprecated path to import EnsembleRegressor. "
+    "Please prefer the new path: "
+    "[from mapie.estimator.regressor import EnsembleRegressor]."
+)
+class EnsembleRegressor(NewEnsembleRegressor):
+    pass

mapie/tests/test_common.py

@@ -280,3 +280,23 @@ def test_warning_when_import_from_old_get_true_label_position():
         from mapie.conformity_scores.utils_classification_conformity_scores\
               import get_true_label_position
         get_true_label_position(np.array([[0.1, 0.2, 0.7]]), np.array([2]))
+
+
+def test_warning_when_import_from_estimator():
+    """Check that a DepreciationWarning is raised when importing from
+    mapie.estimator.estimator"""
+
+    with pytest.warns(
+        FutureWarning, match=r".*WARNING: Deprecated path to import.*"
+    ):
+        from mapie.estimator.estimator import EnsembleRegressor
+        EnsembleRegressor(
+            estimator=LinearRegression(),
+            method="naive",
+            cv=3,
+            agg_function="mean",
+            n_jobs=1,
+            random_state=0,
+            test_size=0.2,
+            verbose=0,
+        )