o
    1 ið  ã                   @   sZ   d dl mZmZmZ d dlmZ erd dlmZ dZedddeee	e
f  fdd	„ƒZd
S )é    )ÚDictÚListÚTYPE_CHECKING)Ú	PublicAPI)ÚAlgorithmConfigg      à?Úconfigr   Úreturnc                    sJ   | j dkr#| j}| d|  dd¡¡}| dt¡‰ ‡ fdd„t|ƒD ƒS g S )NZdatasetÚparallelismZnum_env_runnersé   Znum_cpus_per_read_taskc                    s   g | ]}d ˆ i‘qS )ZCPU© )Ú.0Ú_©Zcpus_per_taskr   úf/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/ray/rllib/offline/resource.pyÚ
<listcomp>   s    z3get_offline_io_resource_bundles.<locals>.<listcomp>)Zinput_Úinput_configÚgetÚDEFAULT_NUM_CPUS_PER_TASKÚrange)r   r   r	   r   r   r   Úget_offline_io_resource_bundles
   s   
ÿr   N)Útypingr   r   r   Zray.rllib.utils.annotationsr   Z%ray.rllib.algorithms.algorithm_configr   r   ÚstrÚfloatr   r   r   r   r   Ú<module>   s    ÿþ