ENH : take into account GMO comments

Vincent Blot · Vincent Blot · commit 3b638e69554f · 2021-11-15T14:14:12.000+01:00
diff --git a/mapie/_typing.py b/mapie/_typing.py
@@ -6,6 +6,4 @@
 except (AttributeError, ModuleNotFoundError):
     ArrayLike = Union[np.ndarray, List[List[float]]]
 
-NumpyInt = np.int64
-
 __all__ = ["ArrayLike"]
diff --git a/mapie/classification.py b/mapie/classification.py
@@ -673,7 +673,7 @@ def predict(
                     [
                         np.ma.masked_greater_equal(
                             y_pred_proba,
-                            y_pred_proba_last[:, iq].reshape(-1, 1)
+                            y_pred_proba_last[:, iq].reshape(-1, 1) - EPSILON
                         ).mask
                         for iq, _ in enumerate(self.quantiles_)
                     ], axis=2
diff --git a/mapie/tests/test_classification.py b/mapie/tests/test_classification.py
@@ -141,7 +141,7 @@
         [True, False, False],
         [True, False, False],
         [False, True, False],
-        [True, True, False],
+        [True, True, True],
         [False, True, False],
         [False, False, True],
         [False, False, True],

Original file line number	Diff line number	Diff line change
`@@ -673,7 +673,7 @@ def predict(`
`673`	`673`	`[`
`674`	`674`	`np.ma.masked_greater_equal(`
`675`	`675`	`y_pred_proba,`
`676`		`- y_pred_proba_last[:, iq].reshape(-1, 1)`
	`676`	`+ y_pred_proba_last[:, iq].reshape(-1, 1) - EPSILON`
`677`	`677`	`).mask`
`678`	`678`	`for iq, _ in enumerate(self.quantiles_)`
`679`	`679`	`], axis=2`