o
    $i                  	   @   s  d dl Z d dlZd dlmZ d dlmZ d dlmZ edkre  Z	e	j
dddd	 e	 \ZZejejr7d
ndd edddZejeejddejrMdndidejdddejejr^dndddededded dddZe Zede j dS dS )    N)tune)HyperBandScheduler)MyTrainableClass__main__z--smoke-test
store_truezFinish quickly for testing)actionhelp   )num_cpustraining_iteration   )	time_attrmax_thyperband_test   T)	fail_fast)namestopverbosefailure_config   episode_reward_meanmax)num_samplesmetricmode	scheduler
   Z   d   )widthheight)
run_configtune_configparam_spacez!Best hyperparameters found were: )argparserayr   ray.tune.schedulersr   ray.tune.utils.mock_trainabler   __name__ArgumentParserparseradd_argumentparse_known_argsargs_init
smoke_test	hyperbandTuner	RunConfigFailureConfig
TuneConfigrandinttunerfitresultsprintget_best_resultconfig r>   r>   `/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/ray/tune/examples/hyperband_example.py<module>   sD   