o `۷i@svddlmZddlmZddlmZeddGdddeZeddGdd d eZeddGd d d eZd S) )Dict) RayTrainError) PublicAPIalpha) stabilityc@seZdZdZdS)TrainingFailedErrorzException raised when training fails from a `trainer.fit()` call. This is either :class:`ray.train.WorkerGroupError` or :class:`ray.train.ControllerError`. N)__name__ __module__ __qualname____doc__r r Q/home/ubuntu/vllm_env/lib/python3.10/site-packages/ray/train/v2/api/exceptions.pyrsrcs:eZdZdZdedeeefffdd ZddZ Z S)WorkerGroupErrora%Exception raised from the worker group during training. Args: error_message: A human-readable error message describing the training worker failures. worker_failures: A mapping from worker rank to the exception that occurred on that worker during training. error_messageworker_failurescs td|||_||_dS)Nz&Training failed due to worker errors: )super__init___error_messager)selfrr __class__r r rs zWorkerGroupError.__init__cCs|j|j|jffSN)rrrrr r r __reduce__szWorkerGroupError.__reduce__) rr r r strrint Exceptionrr __classcell__r r rr rsrcs.eZdZdZdeffdd ZddZZS)ControllerErrorzException raised when training fails due to a controller error. Args: controller_failure: The exception that occurred on the controller. controller_failurecs*tdt|||_||jdS)Nz)Training failed due to controller error: )rrrrwith_traceback __traceback__)rrrr r r)s  zControllerError.__init__cCs|j|jffSr)rrrr r r r0szControllerError.__reduce__)rr r r rrrrr r rr r!srN) typingr!ray.train.v2._internal.exceptionsrray.util.annotationsrrrrr r r r s