We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
ComparisonReport.get_predictions
1 parent 73c8537 commit 3672f38Copy full SHA for 3672f38
skore/src/skore/_sklearn/_comparison/report.py
@@ -342,7 +342,7 @@ def get_predictions(
342
] = "predict",
343
X: ArrayLike | None = None,
344
pos_label: PositiveLabel | None = _DEFAULT,
345
- ) -> list[ArrayLike | list[ArrayLike]]:
+ ) -> list[ArrayLike] | list[list[ArrayLike]]:
346
"""Get predictions from the underlying reports.
347
348
This method has the advantage to reload from the cache if the predictions
@@ -406,7 +406,7 @@ def get_predictions(
406
>>> print([split_predictions.shape for split_predictions in predictions])
407
[(25,), (25,)]
408
"""
409
- return [
+ return [ # type:ignore
410
report.get_predictions(
411
data_source=data_source,
412
response_method=response_method,
0 commit comments