ray.data.datasource.TFRecordDatasource
ray.data.datasource.TFRecordDatasource#
- class ray.data.datasource.TFRecordDatasource(*args, **kwds)[source]#
PublicAPI (alpha): This API is in alpha and may change before becoming stable.
- __init__()#
Methods
__init__
()create_reader
(**kwargs)Return a Reader for the given read arguments.
do_write
(blocks, metadata, path, dataset_uuid)Creates and returns write tasks for a file-based datasource.
file_extension_filter
()on_write_complete
(write_results, **kwargs)Callback for when a write job completes.
on_write_failed
(write_results, error, **kwargs)Callback for when a write job fails.
prepare_read
(parallelism, **read_args)Deprecated: Please implement create_reader() instead.