Skip to content

Commit 73c8537

Browse files
committed
fix typing on _update_dataset_dropdown
1 parent 6caf87d commit 73c8537

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

skore/src/skore/project/widget.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -207,7 +207,7 @@ def _filter_dataframe(self, ml_task: str, report_type: str) -> pd.DataFrame:
207207
df.columns = [col.removesuffix("_mean") for col in df.columns]
208208
return df
209209

210-
def _get_datasets(self, ml_task: str, report_type: str) -> np.ndarray:
210+
def _get_datasets(self, ml_task: str, report_type: str) -> list[str]:
211211
"""Get the unique datasets from the filtered dataframe.
212212
213213
Parameters
@@ -219,7 +219,7 @@ def _get_datasets(self, ml_task: str, report_type: str) -> np.ndarray:
219219
220220
Returns
221221
-------
222-
np.ndarray
222+
list[str]
223223
The unique datasets.
224224
"""
225225
return self._filter_dataframe(ml_task, report_type)["dataset"].unique()
@@ -421,12 +421,12 @@ def __init__(self, dataframe: pd.DataFrame, seed: int = 0) -> None:
421421
layout=widgets.Layout(width="100%", overflow="hidden"),
422422
)
423423

424-
def _update_dataset_dropdown(self, datasets: np.ndarray) -> None:
424+
def _update_dataset_dropdown(self, datasets: list[str]) -> None:
425425
"""Update the dataset dropdown options.
426426
427427
Parameters
428428
----------
429-
datasets : np.ndarray
429+
datasets : list[str]
430430
The datasets to display in the dropdown.
431431
"""
432432
self._dataset_dropdown.options = datasets

0 commit comments

Comments
 (0)