Method _create_batch_iterator

python/ray/data/iterator.py:219–229

(
        self,
        ref_bundles_iter: Iterator[RefBundle],
        prefetch_bytes_callback: Optional[Callable[[int], None]] = None,
        **kwargs,
    )

        )

    def _create_batch_iterator(
        self,
        ref_bundles_iter: Iterator[RefBundle],
        prefetch_bytes_callback: Optional[Callable[[int], None]] = None,
        **kwargs,
    ) -> BatchIterator:
        return BatchIterator(
            ref_bundles_iter,
            prefetch_bytes_callback=prefetch_bytes_callback,
            **kwargs,
        )

    def _iter_batches(
        self,