ray.data.Dataset.take\_batch ============================ .. currentmodule:: ray.data .. automethod:: Dataset.take_batch