From 127f3447eda13b6090c3728438f33a591cab1e90 Mon Sep 17 00:00:00 2001 From: qroa Date: Fri, 13 Dec 2024 14:01:16 +0100 Subject: [PATCH] Fix intergations test --- mapie_v1/integration_tests/tests/test_regression.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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, }