Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
docs: explain that task.get_dataset passes kwargs
  • Loading branch information
knyazer committed Sep 17, 2024
commit ba8eef41b37987f60f92e7070e4dd3c32a0caaaa
7 changes: 5 additions & 2 deletions openml/tasks/task.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Comment thread
PGijsbers marked this conversation as resolved.
Outdated
"""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(
Expand Down