Skip to content

Commit 3672f38

Browse files
committed
Fix ComparisonReport.get_predictions return type
1 parent 73c8537 commit 3672f38

File tree

1 file changed

+2
-2
lines changed
  • skore/src/skore/_sklearn/_comparison

1 file changed

+2
-2
lines changed

skore/src/skore/_sklearn/_comparison/report.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -342,7 +342,7 @@ def get_predictions(
342342
] = "predict",
343343
X: ArrayLike | None = None,
344344
pos_label: PositiveLabel | None = _DEFAULT,
345-
) -> list[ArrayLike | list[ArrayLike]]:
345+
) -> list[ArrayLike] | list[list[ArrayLike]]:
346346
"""Get predictions from the underlying reports.
347347
348348
This method has the advantage to reload from the cache if the predictions
@@ -406,7 +406,7 @@ def get_predictions(
406406
>>> print([split_predictions.shape for split_predictions in predictions])
407407
[(25,), (25,)]
408408
"""
409-
return [
409+
return [ # type:ignore
410410
report.get_predictions(
411411
data_source=data_source,
412412
response_method=response_method,

0 commit comments

Comments
 (0)