Skip to content

Commit 7a207e4

Browse files
author
Vincent Blot
committed
FIX : typing issue
1 parent b4b75b3 commit 7a207e4

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

mapie/classification.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -374,7 +374,7 @@ def _get_last_score_included(
374374
self,
375375
y_pred_proba_cumsum: ArrayLike,
376376
include_last_label: Optional[Union[bool, str]]
377-
) -> Union[ArrayLike, None]:
377+
) -> ArrayLike:
378378
"""
379379
Return the index of the last included sorted probability
380380
depending if we included the first label over the quantile
@@ -390,7 +390,7 @@ def _get_last_score_included(
390390
391391
Returns
392392
-------
393-
ArrayLike
393+
Optional[ArrayLike]
394394
Index of the last included sorted probability.
395395
"""
396396
if (

0 commit comments

Comments
 (0)