diff --git a/mapie_v1/integration_tests/tests/test_regression.py b/mapie_v1/integration_tests/tests/test_regression.py index 3b3a485c1..cda640fd6 100644 --- a/mapie_v1/integration_tests/tests/test_regression.py +++ b/mapie_v1/integration_tests/tests/test_regression.py @@ -209,6 +209,7 @@ def test_intervals_and_predictions_exact_equality_cross(params_cross): "aggregation_method": "median", "method": "plus", "fit_params": {"sample_weight": sample_weight}, + "ensemble": True, "random_state": RANDOM_STATE, }, }, @@ -218,7 +219,6 @@ def test_intervals_and_predictions_exact_equality_cross(params_cross): "alpha": [0.5, 0.5], "conformity_score": GammaConformityScore(), "agg_function": "mean", - "ensemble": True, "cv": Subsample(n_resamplings=20, replace=True, random_state=RANDOM_STATE), @@ -256,6 +256,7 @@ def test_intervals_and_predictions_exact_equality_cross(params_cross): ), "method": "minmax", "aggregation_method": "mean", + "ensemble": True, "allow_infinite_bounds": True, "random_state": RANDOM_STATE, }