REFACTOR: remove random_state from EnsembleRegressor (not used)

scikit-learn-contrib · Jan 13, 2025 · 0a04bfa · 0a04bfa
1 parent 7736558
commit 0a04bfa
Show file tree

Hide file tree

Showing 4 changed files with 0 additions and 6 deletions.
diff --git a/mapie/estimator/regressor.py b/mapie/estimator/regressor.py
@@ -161,7 +161,6 @@ def __init__(
         cv: Optional[Union[int, str, BaseCrossValidator]],
         agg_function: Optional[str],
         n_jobs: Optional[int],
-        random_state: Optional[Union[int, np.random.RandomState]],
         test_size: Optional[Union[int, float]],
         verbose: int
     ):
@@ -170,7 +169,6 @@ def __init__(
         self.cv = cv
         self.agg_function = agg_function
         self.n_jobs = n_jobs
-        self.random_state = random_state
         self.test_size = test_size
         self.verbose = verbose
 

diff --git a/mapie/regression/regression.py b/mapie/regression/regression.py
@@ -549,7 +549,6 @@ def init_fit(
             cv,
             agg_function,
             self.n_jobs,
-            self.random_state,
             self.test_size,
             self.verbose
         )

diff --git a/mapie/tests/test_common.py b/mapie/tests/test_common.py
@@ -310,7 +310,6 @@ def test_warning_when_import_from_estimator():
             cv=3,
             agg_function="mean",
             n_jobs=1,
-            random_state=0,
             test_size=0.2,
             verbose=0,
         )
diff --git a/mapie/tests/test_regression.py b/mapie/tests/test_regression.py
@@ -751,7 +751,6 @@ def test_aggregate_with_mask_with_invalid_agg_function() -> None:
         KFold(n_splits=5, random_state=None, shuffle=True),
         "nonsense",
         None,
-        random_state,
         0.20,
         False
     )
@@ -1032,7 +1031,6 @@ def test_deprecated_ensemble_regressor_fit_warning() -> None:
         KFold(n_splits=5, random_state=None, shuffle=True),
         "nonsense",
         None,
-        random_state,
         0.20,
         False
     )