get\_dataset\_shard =================== .. currentmodule:: ray.train .. autofunction:: get_dataset_shard