o
    c²“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 )NÚdatasetÚparallelismÚnum_env_runnersé   Únum_cpus_per_read_taskc                    s   g | ]}d ˆ i‘qS )ÚCPU© )Ú.0Ú_©Úcpus_per_taskr   úN/home/ubuntu/.local/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    ÿþ