[Data] Ray read_tfrecords allow ray_remote_args configs #51379
Labels
data
Ray Data-related issues
enhancement
Request for new feature and/or capability
triage
Needs triage (eg: priority, bug/not-bug, and owning component)
Description
For parquet reader, its possible to pass in additional Ray task configs via
ray_remote_args
as the doc suggested here. However, such option is not available for the read_tfrecords loader.Use case
To fine-tune the loader configs, we need to offer
ray_remote_args
as init args for read_tfrecords.The text was updated successfully, but these errors were encountered: