o
    ciU                  
   @   s   d dl Z d dlZd dlmZ d dlmZ G dd deZdd Zdd	 Zd
d Z	e
dkrxe  Zejdddd e \ZZeje	ejde gdejrMdndidejdddeeddeddeddddZe Zede j dS dS )    N)tune)LoggerCallbackc                   @   s   e Zd Zdd ZdS )TestLoggerCallbackc                 K   s   t d| d|  d S )NzTestLogger for trial z: )print)self	iterationtrialstrialresultinfo r   U/home/ubuntu/.local/lib/python3.10/site-packages/ray/tune/examples/logging_example.pyon_trial_result   s   z"TestLoggerCallback.on_trial_resultN)__name__
__module____qualname__r   r   r   r   r   r   
   s    r   c                 C   s   d | j| jS )Nz	{}_{}_123)formattrainable_nametrial_id)r	   r   r   r   trial_str_creator   s   r   c                 C   s&   t d d||  d  d |d  S )Ng?d   )timesleep)stepwidthheightr   r   r   evaluation_fn   s   
r   c                 C   sD   | d | d }}t | d D ]}t|||}t||d qd S )Nr   r   steps)
iterations	mean_loss)ranger   r   report)configr   r   r   intermediate_scorer   r   r   easy_objective   s
   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   r   <module>   sH   

