diff --git a/openml/tasks/task.py b/openml/tasks/task.py index 4ad4cec62..1e8671847 100644 --- a/openml/tasks/task.py +++ b/openml/tasks/task.py @@ -145,9 +145,12 @@ 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) -> datasets.OpenMLDataset: - """Download dataset associated with task.""" - return datasets.get_dataset(self.dataset_id) + def get_dataset(self, **kwargs) -> datasets.OpenMLDataset: + """Download dataset associated with task. + + Accepts the same keyword arguments as the `openml.datasets.get_dataset`. + """ + return datasets.get_dataset(self.dataset_id, **kwargs) def get_train_test_split_indices( self,