Add .list to helpers

Author: vhaldemar

examples/async/tuning/datasets.py

@@ -48,6 +48,11 @@ async def main() -> None:
     print(f"New {bad_dataset=} have a bad status {bad_dataset.status=}")
     await dataset.delete()
 
+    # You could call .list not only on .datasets,
+    # but on .completions helper as well, it will substitute corresponding task_type as a filter
+    async for dataset in sdk.datasets.completions.list():
+        await dataset.delete()
+
     async for dataset in sdk.datasets.list():
         await dataset.delete()
 

examples/sync/tuning/datasets.py

@@ -47,6 +47,11 @@ def main() -> None:
     print(f"New {bad_dataset=} have a bad status {dataset.status=}")
     dataset.delete()
 
+    # You could call .list not only on .datasets,
+    # but on .completions helper as well, it will substitute corresponding task_type as a filter
+    for dataset in sdk.datasets.completions.list():
+        dataset.delete()
+
     for dataset in sdk.datasets.list():
         dataset.delete()
 

src/yandex_cloud_ml_sdk/_datasets/task_types.py

@@ -66,5 +66,9 @@ def draft_from_path(self):
     def list_upload_formats(self):
         return partial(self._domain.list_upload_formats, task_type=self._task_type)
 
+    @property
+    def list(self):
+        return partial(self._domain.list, task_type=self._task_type)
+
     def __repr__(self) -> str:
         return f'{self.__class__.__name__}(task_type={self._task_type})'