
    &`i                     v    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            Zd	S )
    )TYPE_CHECKINGDictList)	PublicAPI)AlgorithmConfigg      ?configr   returnc                     | j         dk    rg| j        }|                    d|                     dd                    }|                    dt                    fdt	          |          D             S g S )Ndatasetparallelismnum_env_runners   num_cpus_per_read_taskc                     g | ]}d iS )CPU ).0_cpus_per_tasks     n/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/rllib/offline/resource.py
<listcomp>z3get_offline_io_resource_bundles.<locals>.<listcomp>   s    CCC1&CCC    )input_input_configgetDEFAULT_NUM_CPUS_PER_TASKrange)r   r   r   r   s      @r   get_offline_io_resource_bundlesr      s     }	!!* #&&}fjjARTU6V6VWW$(($&?
 
 DCCCk0B0BCCCC	r   N)typingr   r   r   ray.rllib.utils.annotationsr   %ray.rllib.algorithms.algorithm_configr   r   strfloatr   r   r   r   <module>r$      s    , , , , , , , , , , 1 1 1 1 1 1 FEEEEEE  	$sEz
     r   