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   W/home/ubuntu/veenaModal/venv/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    