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