
    &`i                     &   d Z ddlZddlZd Zd Zd Zd Zd Zd Zd	 Z	d
 Z
d Zd Zeeeeeee	e
eed
ZddddddddddddZdedefdZi dddddd d!d"d#d$d%d&d'd(d)d*d+d*d,d-d.d/d0d/d1d2d3d4d5d6d7d8d9d8d:d;iZd<efd=Zd>efd?ZdS )@z;Registry of algorithm names for tune.Tuner(trainable=[..]).    Nc                  Z    dd l mc mc m}  | j        | j                                        fS Nr   )ray.rllib.algorithms.apporllib
algorithmsappoAPPOget_default_config)r   s    q/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/rllib/algorithms/registry.py_import_appor      sA    ,,,,,,,,,,,,9di224444    c                  Z    dd l mc mc m}  | j        | j                                        fS r   )ray.rllib.algorithms.bcr   r   bcBCr
   )r   s    r   
_import_bcr      sA    ((((((((((((5"%**,,,,r   c                  Z    dd l mc mc m}  | j        | j                                        fS r   )ray.rllib.algorithms.cqlr   r   cqlCQLr
   )r   s    r   _import_cqlr      A    ************7CG..0000r   c                  Z    dd l mc mc m}  | j        | j                                        fS r   )ray.rllib.algorithms.dqnr   r   dqnDQNr
   )r   s    r   _import_dqnr      r   r   c                  Z    dd l mc mc m}  | j        | j                                        fS r   )ray.rllib.algorithms.dreamerv3r   r   	dreamerv3	DreamerV3r
   )r    s    r   _import_dreamerv3r"      sC    666666666666	 3 F F H HHHr   c                  Z    dd l mc mc m}  | j        | j                                        fS r   )ray.rllib.algorithms.impalar   r   impalaIMPALAr
   )r%   s    r   _import_impalar'   %   A    000000000000=&-::<<<<r   c                  Z    dd l mc mc m}  | j        | j                                        fS r   )ray.rllib.algorithms.iqlr   r   iqlIQLr
   )r+   s    r   _import_iqlr-   +   r   r   c                  Z    dd l mc mc m}  | j        | j                                        fS r   )ray.rllib.algorithms.marwilr   r   marwilMARWILr
   )r0   s    r   _import_marwilr2   1   r(   r   c                  Z    dd l mc mc m}  | j        | j                                        fS r   )ray.rllib.algorithms.ppor   r   ppoPPOr
   )r5   s    r   _import_ppor7   7   r   r   c                  Z    dd l mc mc m}  | j        | j                                        fS r   )ray.rllib.algorithms.sacr   r   sacSACr
   )r:   s    r   _import_sacr<   =   r   r   )
r	   r   r   r   r!   r&   r,   r1   r6   r;   r	   r   r   r   r!   r&   r,   r1   r6   r;   )r	   r   r   r   r!   Impalar,   r&   r1   r6   r;   algreturnc                    | t           v rt          |                      d         S | dk    rddlm} |j        S | dk    rddlm} |S | dk    rddlm} |S | dk    rdd	lm} |S t          d
	                    |                     )Nr   script)script_runner__fake)_MockTrainer__sigmoid_fake_data)_SigmoidFakeData__parameter_tuning)_ParameterTuningTrainerzUnknown algorithm {}.)

ALGORITHMSray.tunerB   ScriptRunnerray.rllib.algorithms.mockrD   rF   rH   	Exceptionformat)r>   rB   rD   rF   rH   s        r   _get_algorithm_classrO   `   s     j#  ##	******))	::::::	%	%	%>>>>>>	$	$	$EEEEEE&&/66s;;<<<r   APPOTF1Policyzappo.appo_tf_policyAPPOTF2PolicyAPPOTorchPolicyzappo.appo_torch_policyCQLTFPolicyzcql.cql_tf_policyCQLTorchPolicyzcql.cql_torch_policyDQNTFPolicyzdqn.dqn_tf_policyDQNTorchPolicyzdqn.dqn_torch_policyImpalaTF1Policyzimpala.impala_tf_policyImpalaTF2PolicyImpalaTorchPolicyzimpala.impala_torch_policyMARWILTF1Policyzmarwil.marwil_tf_policyMARWILTF2PolicyMARWILTorchPolicyzmarwil.marwil_torch_policySACTFPolicyzsac.sac_tf_policySACTorchPolicyzsac.sac_torch_policyPPOTF1Policyzppo.ppo_tf_policyPPOTF2PolicyPPOTorchPolicyzppo.ppo_torch_policypolicy_classc                 R    t          j        dd| j                  }|t          v r|S dS )zReturns a string name for the provided policy class.

    Args:
        policy_class: RLlib policy class, e.g. A3CTorchPolicy, DQNTFPolicy, etc.

    Returns:
        A string name uniquely mapped to the given policy class.
    z_traced$ N)resub__name__POLICIES)rb   names     r   get_policy_class_namerj      s/     6*b,"788Dx4r   ri   c                     | t           vrdS t           |          }t          j        d|z             }t          ||           sdS t	          ||           S )zReturn an actual policy class given the string name.

    Args:
        name: string name of the policy class.

    Returns:
        Actual policy class for the given name.
    Nzray.rllib.algorithms.)rh   	importlibimport_modulehasattrgetattr)ri   pathmodules      r   get_policy_classrr      sZ     8tD>D$%<t%CDDF64   t64   r   )__doc__rl   re   r   r   r   r   r"   r'   r-   r2   r7   r<   rI   ALGORITHMS_CLASS_TO_NAMEstrtyperO   rh   rj   rr    r   r   <module>rx      si   A A     				5 5 5- - -1 1 11 1 1I I I= = =1 1 1= = =1 1 11 1 1 
" 
 
  =c =d = = = =8** / &	
 , & , 0 0 5 0 0 5 & ,  '!" '#$ ,% ,    *!3 ! ! ! ! ! !r   