Data Loading and Checkpointing
==============================

Utilities for streaming training data from Unity Catalog volumes and checkpointing to them.
They are available in GPU environment version 6 and above. Elsewhere, install the ``data`` extra to pull
in their dependencies: ``pip install "databricks-sdk-air[data]"``.

.. currentmodule:: databricks.air.data

UCVolumeDataset
---------------

.. autoclass:: UCVolumeDataset
   :members:

DataLoader
----------

.. autoclass:: DataLoader
   :members:

UCVolumeWriter
--------------

.. autoclass:: UCVolumeWriter
   :members:

UCVolumeReader
--------------

.. autoclass:: UCVolumeReader
   :members:

Checkpointable
--------------

.. autoclass:: Checkpointable
   :members: