&`iddlZddlZddlZddlmZmZmZmZmZm Z ddl m Z m Z ddl mZddlmZmZmZmZmZmZmZejeZerddlmZeeefZde ee eee e fdeefd Zddddddd deee d ed d eedeedeedeedeeedee fdZ dee fdZ!dS)N) TYPE_CHECKING CollectionListOptionalTypeUnion)Callback CallbackList)RAY_TUNE_CALLBACKS_ENV_VAR) CSVLoggerCSVLoggerCallback JsonLoggerJsonLoggerCallbackLegacyLoggerCallback TBXLoggerTBXLoggerCallback) AirVerbosity callbacksreturncBg}|D]}|t|jz }|SN)list_SAVED_FILE_TEMPLATES)r templatescallbacks k/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/tune/utils/callback.py%_get_artifact_templates_for_callbacksrs5I::T(8999 ) air_verbosity entrypointmetricmodeconfigprogress_metricsrrr r!r"r#r$c ^|pg}t}||d}d} d} ddlmt fd|D} | r,|*t dfd|D} | }|.ddlm} | |d ||||| }| |n$| s"|| }| |t|D]\}}t|tr1t|jvrd }t|jvrd } t |jvrd } Kt|t"rd }ct|t$rd } {t|t&rd } t(jd ddkr|s!| t#| s!| t%| sM | t'n*#t.$rt dYnwxYw|S)aCreate default callbacks for `Tuner.fit()`. This function takes a list of existing callbacks and adds default callbacks to it. Specifically, three kinds of callbacks will be added: 1. Loggers. Ray Tune's experiment analysis relies on CSV and JSON logging. 2. Syncer. Ray Tune synchronizes logs and checkpoint between workers and the head node. 2. Trial progress reporter. For reporting intermediate progress, like trial results, Ray Tune uses a callback. These callbacks will only be added if they don't already exist, i.e. if they haven't been passed (and configured) by the user. A notable case is when a Logger is passed, which is not a CSV or JSON logger - then a CSV and JSON logger will still be created. Lastly, this function will ensure that the Syncer callback comes after all Logger callbacks, to ensure that the most up-to-date logs and checkpoints are synced across nodes. Fr)TrialProgressCallbackc38K|]}t|VdSr isinstance.0cr&s r z,_create_default_callbacks..Ts?&&12 1+,,&&&&&&rNz?AIR_VERBOSITY is set, ignoring passed-in TrialProgressCallback.c4g|]}t||Sr(r*s r z-_create_default_callbacks..\s9    16K(L(L    r)_detect_reporter) num_samplesr r!r"r#r$)r!r$T"TUNE_DISABLE_AUTO_CALLBACK_LOGGERS01zThe TensorboardX logger cannot be instantiated because either TensorboardX or one of it's dependencies is not installed. Please make sure you have the latest version of TensorboardX installed: `pip install -U tensorboardx`)_initialize_env_callbacksextendray.tune.progress_reporterr&anyloggerwarningray.tune.experimental.outputr1append enumerater)rr logger_classesrrr rrosenvironget ImportError)rrr r!r"r#r$ env_callbackshas_csv_loggerhas_json_loggerhas_tbx_loggerhas_trial_progress_callback new_callbacks_detect_air_reporterair_progress_reportertrial_progress_callbackirr&s @r_create_default_callbacksrO'sBRI.//M ]###NON@@@@@@"%&&&&6?&&&###"}'@ M           "        !5 4 !-! ! !  .//// (2"7"7,<# # #  0111!++ " " 8 h 4 5 5 "H333!%X444"&H333!% "3 4 4 "!NN "4 5 5 ""OO "3 4 4 "!N z~~:C@@CGG 2   .00 1 1 1 3   /11 2 2 2    !2!4!45555   O  s!!H$H*)H*c ^g}tjtd}|s|S|dD]}|}|s |dd\}}tj|}t||}t|ts'td|dt|j|}||#t t"t$tf$r}t%d|d|d }~wwxYw|S) zInitialize callbacks from environment variable. Returns: List of callbacks initialized from environment variable. ,.r2zCallback class 'z&' must be a subclass of Callback, got z Failed to import callback from ''N)rArBrCr splitstriprsplit importlib import_modulegetattr issubclassr TypeErrortype__name__r>rDAttributeError ValueError) r callbacks_str callback_path module_path class_namemodule callback_clsres rr7r7sn IJNN#=rBBM &,,S11YY %++--    Y&3&:&:3&B&B #K,[99F"6:66LlH55 C}CC%),%7%7%@CC$|~~H   X & & & &^ZC Y Y YP PPPQQWX X Y sBC44D*D%%D*)"rXloggingrAtypingrrrrrrray.tune.callbackr r ray.tune.constantsr ray.tune.loggerr r rrrrr getLoggerr^r;r=rDEFAULT_CALLBACK_CLASSESstrrdictrOr7r/rrrqs< IIIIIIIIIIIIIIII44444444999999  8 $ $:999999T(^T$x.%9<GH #Y/3 $ !26oooX'oN+o o SM o 3- o TNoz#/o (^ooood4>r