Now the linting tests pass..

scikit-learn-contrib · Nov 16, 2022 · cdde3c5 · cdde3c5
1 parent ddaec0c
commit cdde3c5
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/mapie/tests/test_classification.py b/mapie/tests/test_classification.py
@@ -23,7 +23,7 @@
 from mapie._typing import ArrayLike, NDArray
 
 
-METHODS = ["score", "cumulated_score", "raps","mondrian"]
+METHODS = ["score", "cumulated_score", "raps", "mondrian"]
 WRONG_METHODS = ["scores", "cumulated", "test", "", 1, 2.5, (1, 2)]
 WRONG_INCLUDE_LABELS = ["randomised", "True", "False", "other", 1, 2.5, (1, 2)]
 Y_PRED_PROBA_WRONG = [
@@ -888,7 +888,7 @@ def test_toy_dataset_predictions(strategy: str) -> None:
         alpha=0.5,
         include_last_label=args_predict["include_last_label"],
         agg_scores=args_predict["agg_scores"]
-    ) 
+    )
 
     np.testing.assert_allclose(y_ps[:, :, 0], y_toy_mapie[strategy])
     np.testing.assert_allclose(

diff --git a/mapie/utils.py b/mapie/utils.py
@@ -616,7 +616,9 @@ def check_alpha_and_last_axis(vector: NDArray, alpha_np: NDArray):
         return vector, alpha_np
 
 
-def compute_quantiles(vector: NDArray, alpha: NDArray,mondrian = False) -> NDArray:
+def compute_quantiles(vector: NDArray,
+                      alpha: NDArray,
+                      mondrian=False) -> NDArray:
     """Compute the desired quantiles of a vector.
 
     Parameters