diff --git a/openml/tasks/task.py b/openml/tasks/task.py index 1e8671847..064b834ba 100644 --- a/openml/tasks/task.py +++ b/openml/tasks/task.py @@ -145,7 +145,7 @@ def _get_repr_body_fields(self) -> Sequence[tuple[str, str | int | list[str]]]: ] return [(key, fields[key]) for key in order if key in fields] - def get_dataset(self, **kwargs) -> datasets.OpenMLDataset: + def get_dataset(self, **kwargs: Any) -> datasets.OpenMLDataset: """Download dataset associated with task. Accepts the same keyword arguments as the `openml.datasets.get_dataset`.