&`iVddlmZddlmZmZmZddlmZmZddl m Z ddl m Z dZ e GddeeZe Gd d eeZe Gd d eZe Gd deZe GddeZGddeZe GddeZe GddeZe GddeZe GddeZe GddeZe GddeZdS) )Enum)DictListOptional) BaseModelField) JobDetails) DeveloperAPIiPc&eZdZdZdZdZdZdZdZdS) RunStatusEnumz*Enumeration for the status of a train run.STARTEDRUNNINGFINISHEDERROREDABORTEDN) __name__ __module__ __qualname____doc__r rrrrt/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/train/_internal/state/schema.pyr r s/44GGHGGGGrr ceZdZdZdZdS)ActorStatusEnumDEADALIVEN)rrrrrrrrrrs D EEErrceZdZUdZedZeed<edZe ed<edZ e ed<ed Z e ed <ed Z eed <ed Z eed<edZe ed<edZee ed<edZeed<edZeeefed<dS)TrainWorkerInfoMetadata of a Ray Train worker.zActor ID of the worker. descriptionactor_idzWorld rank of the worker. world_rankzLocal rank of the worker. local_rankzNode rank of the worker. node_rankz-ID of the node that the worker is running on.node_idz5IP address of the node that the worker is running on.node_ipzProcess ID of the worker.pidz+A list of GPU ids allocated to that worker.gpu_idsz>The status of the train worker actor. It can be ALIVE or DEAD.status&The resources allocated to the worker. resourcesN)rrrrrr"str__annotations__r#intr$r%r&r'r(r)rr*rr,rfloatrrrrr"s))E&?@@@Hc@@@e(CDDDJDDDe(CDDDJDDDU'ABBBIsBBB5%TUUUGSUUU5KGSu!<===C===AGT#Y$eTFO#(%<###ItCJrrcPeZdZUeed<eed<eeed<eeed<dS) MemoryInforssvmspfaultspageinsN)rrrr/r.rrrrr2r2:sD HHH HHH c] c]rr2cFeZdZUeed<eeeed<eed<dS) ProcessStats cpuPercentmem memoryInfoN) rrrr0r.rrr/r2rrrr8r8Bs? $s) rr8c$eZdZUeed<eed<dS)ProcessGPUUsager(gpuMemoryUsageN)rrrr/r.rrrr=r=Js' HHHrr=cbeZdZUeed<eed<eed<eeed<eed<eed<eed<dS) GPUStatsuuidindexnameutilizationGpu memoryUsed memoryTotal processInfoN) rrrr-r.r/rr0r=rrrr@r@Psa III JJJ IIIUO###      rr@cteZdZUdZeddZeeed<ee dZ e e ed<dS) TrainWorkerInfoWithDetailsrNzProcess stats of the worker.r processStatszSGPU stats of the worker. Only returns GPUs that are attached to the worker process.)default_factoryr!gpus) rrrrrrJrr8r.listrLrr@rrrrIrI[s))+05 8,,,L(<(!5 ID$x.rrIceZdZUedZeed<edZeed<edZe eed<dS) TrainDatasetInfoz;The key of the dataset dict specified in Ray Train Trainer.r rCzThe uuid of the dataset. dataset_uuidzThe name of the dataset. dataset_nameN) rrrrrCr-r.rPrQrrrrrOrOks~QD#*DEEEL#EEE"'%4N"O"O"OL(3-OOOOOrrOceZdZUdZedZeed<edZeed<edZ eed<ed Z eed <ed Z e e ed <ed Ze eed<edZeed<edZeed<edZeed<edZeeed<edZe eeefed<dS) TrainRunInfo?Metadata for a Ray Train run and information about its workers.zThe name of the Train run.r rCz)The unique identifier for each Train run.idzThe Ray Job ID.job_idz!Actor Id of the Train controller.controller_actor_id/A List of Train workers sorted by global ranks.workersz*A List of dataset info for this Train run.datasetszlThe current status of the train run. It can be one of the following: RUNNING, FINISHED, ERRORED, or ABORTED. run_statuszJDetailed information about the current run status, such as error messages. status_detailz7The UNIX timestamp of the start time of this Train run. start_time_msz_The UNIX timestamp of the end time of this Train run. If null, the Train run has not ended yet. end_time_msr+r,N)rrrrrrCr-r.rUrVrWrYrrrZrOr[r r\r]r/r^rr,rr0rrrrSrStsII">???D#???e KLLLBLLL%$5666FC666$u1TUUUUUU%*UE&&&GT/ "(-u@(((Hd#$!&=!!!J "M3MM3"'4"""K#).<)))ItDe$%rrScreZdZUdZedZeeed<eddZ e e ed<dS)TrainRunInfoWithDetailsrTrXr rYNz/Details of the job that started this Train run. job_details) rrrrrrYrrIr.rarr rrrr`r`sII05E111GT, -). K)))K*%rr`c&eZdZUeeed<dS)TrainRunsResponse train_runsN)rrrrr`r.rrrrcrcs$,------rrcN)enumrtypingrrrray._common.pydantic_compatrr)ray.dashboard.modules.job.pydantic_modelsr ray.util.annotationsr MAX_ERROR_STACK_TRACE_LENGTHr-r rrr2r8r=r@rIrOrSr`rcrrrrks ''''''''''88888888@@@@@@------$     C    c4 i.9i !!!!!y!!!        PPPPPyPPP9Bl..... .....r