Skip to content

Commit caa9de6

Browse files
author
mata
committed
Now the linting tests pass..
1 parent 1b0e00d commit caa9de6

File tree

2 files changed

+5
-3
lines changed

2 files changed

+5
-3
lines changed

mapie/tests/test_classification.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@
2222
from mapie.metrics import classification_coverage_score
2323
from mapie.utils import check_alpha
2424

25-
METHODS = ["score", "cumulated_score", "raps","mondrian"]
25+
METHODS = ["score", "cumulated_score", "raps", "mondrian"]
2626
WRONG_METHODS = ["scores", "cumulated", "test", "", 1, 2.5, (1, 2)]
2727
WRONG_INCLUDE_LABELS = ["randomised", "True", "False", "other", 1, 2.5, (1, 2)]
2828
Y_PRED_PROBA_WRONG = [
@@ -887,7 +887,7 @@ def test_toy_dataset_predictions(strategy: str) -> None:
887887
alpha=0.5,
888888
include_last_label=args_predict["include_last_label"],
889889
agg_scores=args_predict["agg_scores"]
890-
)
890+
)
891891

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

mapie/utils.py

+3-1
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)