o `۷i@s@ddlmZddlmZmZmZddlmZmZddl m Z ddl m Z dZ e GdddeeZe Gd d d eeZe Gd d d eZe Gd ddeZe GdddeZGdddeZe GdddeZe GdddeZe GdddeZe GdddeZe GdddeZe GdddeZdS) )Enum)DictListOptional) BaseModelField) JobDetails) DeveloperAPIiPc@s$eZdZdZdZdZdZdZdZdS) RunStatusEnumz*Enumeration for the status of a train run.STARTEDRUNNINGFINISHEDERROREDABORTEDN) __name__ __module__ __qualname____doc__r r r rrrrV/home/ubuntu/vllm_env/lib/python3.10/site-packages/ray/train/_internal/state/schema.pyr sr c@seZdZdZdZdS)ActorStatusEnumDEADALIVEN)rrrrrrrrrrsrc@seZdZUdZeddZeed<eddZe ed<eddZ e ed<ed dZ e ed <ed dZ eed <ed dZ eed<eddZe ed<eddZee ed<eddZeed<eddZeeefed<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)rrrrrrstr__annotations__rintrr r!r"r#r$rr%rr'rfloatrrrrr"s( rc@s6eZdZUeed<eed<eeed<eeed<dS) MemoryInforssvmspfaultspageinsN)rrrr*r)rrrrrr,:s  r,c@s.eZdZUeed<eeeed<eed<dS) ProcessStats cpuPercentmem memoryInfoN) rrrr+r)rrr*r,rrrrr1Bs  r1c@seZdZUeed<eed<dS)ProcessGPUUsager#gpuMemoryUsageN)rrrr*r)rrrrr5Js  r5c@sJeZdZUeed<eed<eed<eeed<eed<eed<eed<dS) GPUStatsuuidindexnameutilizationGpu memoryUsed memoryTotal processInfoN) rrrr(r)r*rr+r5rrrrr7Ps   r7c@sBeZdZUdZedddZeeed<ee ddZ 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_factoryrgpus) rrrrrr@rr1r)listrBrr7rrrrr?[s r?c@sHeZdZUeddZeed<eddZeed<eddZe eed<dS) TrainDatasetInfoz;The key of the dataset dict specified in Ray Train Trainer.rr:zThe uuid of the dataset. dataset_uuidzThe name of the dataset. dataset_nameN) rrrrr:r(r)rErFrrrrrrDks rDc@seZdZUdZeddZeed<eddZeed<eddZ eed<ed dZ eed <ed dZ e e ed <ed dZe eed<eddZeed<eddZeed<eddZeed<eddZeeed<eddZe eeefed<dS) TrainRunInfo?Metadata for a Ray Train run and information about its workers.zThe name of the Train run.rr:z)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)rrrrrr:r(r)rIrJrKrMrrrNrDrOr rPrQr*rRrr'rr+rrrrrGts6 rGc@s@eZdZUdZeddZeeed<edddZ e e ed<dS)TrainRunInfoWithDetailsrHrLrrMNz/Details of the job that started this Train run. job_details) rrrrrrMrr?r)rTrrrrrrrSs rSc@seZdZUeeed<dS)TrainRunsResponse train_runsN)rrrrrSr)rrrrrUs rUN)enumrtypingrrrray._common.pydantic_compatrr)ray.dashboard.modules.job.pydantic_modelsrray.util.annotationsr MAX_ERROR_STACK_TRACE_LENGTHr(r rrr,r1r5r7r?rDrGrSrUrrrrs:    !