ray/doc/source/data/custom-data.rst

17 lines
567 B
ReStructuredText
Raw Normal View History

.. _datasets_custom_datasource:
------------------------
Using Custom Datasources
------------------------
Datasets can read and write in parallel to `custom datasources <package-ref.html#custom-datasource-api>`__ defined in Python.
Once you have implemented `YourCustomDataSource`, you can use it like any other source in Ray Data:
.. code-block:: python
# Read from a custom datasource.
ds = ray.data.read_datasource(YourCustomDatasource(), **read_args)
# Write to a custom datasource.
ds.write_datasource(YourCustomDatasource(), **write_args)