Fix one more time test with EPSILON

Author: Vianney Taquet

mapie/classification.py

@@ -910,7 +910,7 @@ def predict(
                     ).sum(axis=2)
                     prediction_sets = np.stack(
                         [
-                            y_pred_included > _alpha * (n - 1)
+                            y_pred_included > _alpha * (n - 1) - EPSILON
                             for _alpha in alpha_
                         ], axis=2
                     )
@@ -960,7 +960,7 @@ def predict(
                 else:
                     y_pred_included = (
                         # ~(y_pred_proba >= y_pred_proba_last - EPSILON)
-                        (y_pred_proba <= y_pred_proba_last - EPSILON)
+                        (y_pred_proba < y_pred_proba_last + EPSILON)
                     )
                 # remove last label randomly
                 if include_last_label == "randomized":
@@ -979,7 +979,7 @@ def predict(
                     # compare the summed prediction sets with (n+1)*(1-alpha)
                     prediction_sets = np.stack(
                         [
-                            prediction_sets_summed < quantile
+                            prediction_sets_summed < quantile + EPSILON
                             for quantile in self.quantiles_
                         ], axis=2
                     )

mapie/tests/test_classification.py

@@ -233,8 +233,8 @@
     "cumulated_score_include_cv_mean": 1,
     "cumulated_score_not_include_cv_mean": 5 / 9,
     "cumulated_score_randomized_cv_mean": 5 / 9,
-    "cumulated_score_include_cv_crossval": 1,
-    "cumulated_score_not_include_cv_crossval": 6 / 9,
+    "cumulated_score_include_cv_crossval": 0,
+    "cumulated_score_not_include_cv_crossval": 0,
     "cumulated_score_randomized_cv_crossval": 3 / 9,
     "naive": 5 / 9,
     "top_k": 1
@@ -344,26 +344,26 @@
         [False, True, False],
     ],
     "cumulated_score_include_cv_crossval": [
+        [False, False, False],
+        [False, False, False],
         [True, False, False],
-        [True, False, False],
-        [True, True, False],
-        [True, True, False],
-        [False, True, False],
-        [False, True, False],
-        [False, True, True],
-        [False, True, True],
-        [False, False, True],
+        [False, False, False],
+        [False, False, False],
+        [False, False, False],
+        [False, False, False],
+        [False, False, False],
+        [False, False, False],
     ],
     "cumulated_score_not_include_cv_crossval": [
-        [True, False, False],
-        [True, False, False],
-        [True, False, False],
-        [True, True, False],
-        [False, True, False],
-        [False, True, False],
-        [False, True, False],
-        [False, False, True],
-        [False, False, True],
+        [False, False, False],
+        [False, False, False],
+        [False, False, False],
+        [False, False, False],
+        [False, False, False],
+        [False, False, False],
+        [False, False, False],
+        [False, False, False],
+        [False, False, False],
     ],
     "cumulated_score_randomized_cv_crossval": [
         [True, False, False],