docs: explain that task.get_dataset passes kwargs

openml · Sep 17, 2024 · ba8eef4 · ba8eef4
1 parent 2f36b07
commit ba8eef4
Showing 1 changed file with 5 additions and 2 deletions.
diff --git 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(