MTN: clarify public/private API in __init__ modules (#678)

Valentin-Laurent · web-flow · commit e67b3dbe59bd · 2025-05-06T13:58:07.000+02:00
diff --git a/mapie/__init__.py b/mapie/__init__.py
@@ -1,10 +1,21 @@
-from . import classification, metrics, regression, utils
+from . import (
+    classification,
+    metrics,
+    regression,
+    utils,
+    multi_label_classification,
+    calibration,
+    subsample,
+)
 from ._version import __version__
 
 __all__ = [
     "regression",
     "classification",
+    "multi_label_classification",
+    "calibration",
     "metrics",
     "utils",
+    "subsample",
     "__version__"
 ]
diff --git a/mapie/estimator/__init__.py b/mapie/estimator/__init__.py
@@ -1,7 +0,0 @@
-from .regressor import EnsembleRegressor
-from .classifier import EnsembleClassifier
-
-__all__ = [
-    "EnsembleRegressor",
-    "EnsembleClassifier",
-]
