Skip to content

Commit cdde3c5

Browse files
author
mata
committed
Now the linting tests pass..
1 parent ddaec0c commit cdde3c5

File tree

2 files changed

+5
-3
lines changed

2 files changed

+5
-3
lines changed

mapie/tests/test_classification.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@
2323
from mapie._typing import ArrayLike, NDArray
2424

2525

26-
METHODS = ["score", "cumulated_score", "raps","mondrian"]
26+
METHODS = ["score", "cumulated_score", "raps", "mondrian"]
2727
WRONG_METHODS = ["scores", "cumulated", "test", "", 1, 2.5, (1, 2)]
2828
WRONG_INCLUDE_LABELS = ["randomised", "True", "False", "other", 1, 2.5, (1, 2)]
2929
Y_PRED_PROBA_WRONG = [
@@ -888,7 +888,7 @@ def test_toy_dataset_predictions(strategy: str) -> None:
888888
alpha=0.5,
889889
include_last_label=args_predict["include_last_label"],
890890
agg_scores=args_predict["agg_scores"]
891-
)
891+
)
892892

893893
np.testing.assert_allclose(y_ps[:, :, 0], y_toy_mapie[strategy])
894894
np.testing.assert_allclose(

mapie/utils.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -616,7 +616,9 @@ def check_alpha_and_last_axis(vector: NDArray, alpha_np: NDArray):
616616
return vector, alpha_np
617617

618618

619-
def compute_quantiles(vector: NDArray, alpha: NDArray,mondrian = False) -> NDArray:
619+
def compute_quantiles(vector: NDArray,
620+
alpha: NDArray,
621+
mondrian=False) -> NDArray:
620622
"""Compute the desired quantiles of a vector.
621623
622624
Parameters

0 commit comments

Comments
 (0)