
    &`iU                     2   d dl Z d dlZd dlmZ d dlmZ  G d de          Zd Zd Zd Z	e
d	k    r e j                    Ze                    d
dd           e                                \  ZZ ej        e	 ej        d e            gdej        rdndi           ej        dddee          d ej        dd           ej        dd          d          Ze                                Z ede                                j                   dS dS )    N)tune)LoggerCallbackc                       e Zd Zd ZdS )TestLoggerCallbackc                 0    t          d| d|            d S )NzTestLogger for trial z: )print)self	iterationtrialstrialresultinfos         u/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/tune/examples/logging_example.pyon_trial_resultz"TestLoggerCallback.on_trial_result   s'    7e77v7788888    N)__name__
__module____qualname__r    r   r   r   r   
   s#        9 9 9 9 9r   r   c                 B    d                     | j        | j                  S )Nz	{}_{}_123)formattrainable_nametrial_id)r   s    r   trial_str_creatorr      s    e2ENCCCr   c                 R    t          j        d           d|| z  dz  z   dz  |dz  z   S )Ng?d   )timesleep)stepwidthheights      r   evaluation_fnr#      s1    JsOOO%$,$$"-<<r   c                     | d         | d         }}t          | d                   D ]*}t          |||          }t          j        ||d           +d S )Nr!   r"   steps)
iterations	mean_loss)ranger#   r   report)configr!   r"   r    intermediate_scores        r   easy_objectiver,      sk    7OVH%56EfWo&& K K*4??46HIIJJJJ	K Kr   __main__z--smoke-test
store_truezFinish quickly for testing)actionhelphyperband_testtraining_iteration   r   )name	callbacksstopr'   min   )metricmodenum_samplestrial_name_creatortrial_dirname_creator
   )r%   r!   r"   )
run_configtune_configparam_spacezBest hyperparameters: )argparser   rayr   ray.tune.loggerr   r   r   r#   r,   r   ArgumentParserparseradd_argumentparse_known_argsargs_Tuner	RunConfig
smoke_test
TuneConfigrandint
loguniformtunerfitresultsr   get_best_resultr*   r   r   r   <module>rU      s           * * * * * *9 9 9 9 9 9 9 9
D D D= = =
K K K z$X$&&F
|2N     %%''GD!DJ!4>!))++,&T_(E#F
 
 

 $DO0"3
 
 
 !T\"c**%dob#..
 
  E( iikkG	E
"G$;$;$=$=$DEEEEE; r   