o
    `۷i[                     @   s:   d dl mZ d dlmZ dee dedee fddZdS )	    )Iterable)Blockblocksper_task_row_limitreturnc           	      c   s~    d}| D ]7}||kr dS ddl m} ||}| }|| |kr+|V  ||7 }q|| }|jd|dd}|V   dS dS )zDIterate over blocks, accumulating rows up to the per-task row limit.r   )BlockAccessorT)copyN)ray.data.blockr   	for_blocknum_rowsslice)	r   r   	rows_readblockr   accessor
block_rowsremaining_rowssliced_block r   N/home/ubuntu/vllm_env/lib/python3.10/site-packages/ray/data/datasource/util.py_iter_sliced_blocks   s    

r   N)typingr   r	   r   intr   r   r   r   r   <module>   s    