&`iddlmZmZmZmZddlmZddlmZddl m Z e GddZ e Gdde Z d S) )AnyDictListOptional) Checkpoint)TrainRunContext) DeveloperAPIceZdZdZdS)RayTrainCallbackz"Base Ray Train callback interface.N)__name__ __module__ __qualname____doc__m/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/train/v2/api/callback.pyr r s,,Drr cneZdZdZdedeeeefde e fdZ dedee e ffdZdS) UserCallbackzCallback interface for custom user-defined callbacks to handling events during training. This callback is called on the Ray Train controller process, not on the worker processes. run_contextmetrics checkpointcdS)a*Called after all workers have reported a metric + checkpoint via `ray.train.report`. Args: run_context: The `TrainRunContext` for the current training run. metrics: A list of metric dictionaries reported by workers, where metrics[i] is the metrics dict reported by worker i. checkpoint: A Checkpoint object that has been persisted to storage. This is None if no workers reported a checkpoint (e.g. `ray.train.report(metrics, checkpoint=None)`). Nr)selfrrrs r after_reportzUserCallback.after_reports " rworker_exceptionscdS)aCalled after one or more workers have raised an exception. Args: run_context: The `TrainRunContext` for the current training run. worker_exceptions: A dict from worker world rank to the exception raised by that worker. Nr)rrrs rafter_exceptionzUserCallback.after_exception+s  rN)r r rrrrrstrrrrrint Exceptionrrrrrrs $ d38n% Z(     &  *  ?CCN?S       rrN) typingrrrr ray.trainr(ray.train.v2._internal.execution.contextrray.util.annotationsr r rrrrr%s,,,,,,,,,,,, DDDDDD------        % % % % % #% % % % % r