docs(sdk): document the dataset CLI and Python API

python/docs/cli/dataset.rst

@@ -0,0 +1,8 @@
+Dataset
++++++++
+
+.. lightning-reference:: dataset
+   :root-label: lightning dataset
+   :anchor-prefix: dataset
+
+   from lightning_sdk.cli.groups import dataset

python/docs/cli/index.rst

@@ -61,6 +61,7 @@ Common Workflows
 * Submit and inspect training or batch work with :doc:`job` and :doc:`mmt`.
 * Build and operate inference services with :doc:`deployment` and :doc:`model`.
 * Move data and artifacts with :doc:`file`, :doc:`folder`, :doc:`container`, and :doc:`cp`.
+* Upload and download versioned datasets with :doc:`dataset`.
 * Configure accounts, organizations, teamspaces, cloud accounts, and SSH with
   :doc:`config`, :doc:`api-key`, and :doc:`ssh`.
 * Manage lower-level sandbox sessions with :doc:`sandbox`.
@@ -84,6 +85,7 @@ reference.
    api-key
    file
    folder
+   dataset
    ssh
    studio
    sandbox

python/docs/sdk/api.rst

@@ -86,4 +86,5 @@ Reference Pages
    api/user
    api/status
    api/models
+   api/datasets
    api/experiment

python/docs/sdk/api/datasets.rst

@@ -0,0 +1,17 @@
+Datasets
+========
+
+Download and upload datasets programmatically. These are the Python
+equivalents of the ``lightning dataset`` CLI commands.
+
+.. currentmodule:: lightning_sdk.datasets
+
+.. autofunction:: download_dataset
+
+.. autofunction:: upload_dataset
+
+.. autoclass:: DownloadedDatasetInfo
+   :members:
+
+.. autoclass:: UploadedDatasetInfo
+   :members: