From ba8eef41b37987f60f92e7070e4dd3c32a0caaaa Mon Sep 17 00:00:00 2001 From: Roman Knyazhitskiy Date: Tue, 17 Sep 2024 21:49:55 +0200 Subject: [PATCH] docs: explain that ```task.get_dataset``` passes kwargs --- openml/tasks/task.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/openml/tasks/task.py b/openml/tasks/task.py index 358719053..71f8c774a 100644 --- a/openml/tasks/task.py +++ b/openml/tasks/task.py @@ -145,8 +145,11 @@ 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: - """Download dataset associated with task.""" + 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(