o
    `Û·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 )é    )ÚTYPE_CHECKINGÚDictÚList)Ú	PublicAPI)ÚAlgorithmConfigg      à?Úconfigr   Úreturnc                    sJ   | j dkr#| j}| d|  dd¡¡}| dt¡‰ ‡ fdd„t|ƒD ƒS g S )NÚdatasetÚparallelismÚnum_env_runnersé   Únum_cpus_per_read_taskc                    s   g | ]}d ˆ i‘qS )ÚCPU© )Ú.0Ú_©Úcpus_per_taskr   úP/home/ubuntu/vllm_env/lib/python3.10/site-packages/ray/rllib/offline/resource.pyÚ
<listcomp>   s    z3get_offline_io_resource_bundles.<locals>.<listcomp>)Ú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   Úray.rllib.utils.annotationsr   Ú%ray.rllib.algorithms.algorithm_configr   r   ÚstrÚfloatr   r   r   r   r   Ú<module>   s    ÿþ