Fix intergations test

scikit-learn-contrib · Dec 13, 2024 · 127f344 · 127f344
1 parent b53ff0d
commit 127f344
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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,
         }