o ci"@s~ddlmZddlmZmZmZddlmZmZddl m Z ddl m Z dZ e GdddeeZe Gd d d eeZe Gd d d eeZe Gd ddeZe GdddeZe GdddeZe GdddeZGdddeZe GdddeZe GdddeZe GdddeZe GdddeZe Gdd d eZe Gd!d"d"eZe Gd#d$d$eZd%S)&)Enum)DictListOptional) BaseModelField) JobDetails) DeveloperAPIiPc@s0eZdZdZdZdZdZdZdZdZ dZ d Z d S) RunStatusz5Enumeration of the possible statuses for a Train run. INITIALIZING SCHEDULINGRUNNING RESTARTINGRESIZINGFINISHEDERROREDABORTEDN) __name__ __module__ __qualname____doc__r r r rrrrrrrW/home/ubuntu/.local/lib/python3.10/site-packages/ray/train/v2/_internal/state/schema.pyr sr c@s$eZdZdZdZdZdZdZdZdS)RunAttemptStatusz=Enumeration of the possible statuses for a Train run attempt.PENDINGr rrrN) rrrrrr rrrrrrrr$src@seZdZdZdZdZdS) ActorStatusz5Enumeration of the statuses for a Train worker actor.ALIVEDEADN)rrrrrrrrrrr7src@s,eZdZUdZeddZeeefe d<dS)TrainResourcesz.Resources allocated for a Train worker or run.zVA dictionary specifying the types and amounts of resources allocated (e.g., CPU, GPU). description resourcesN) rrrrrr!rstrfloat__annotations__rrrrrAs rc@seZdZUdZeddZeed<eddZeed<eddZ eed<ed dZ e ed <ed dZ e ed <ed dZ e ed<eddZeed<eddZeeed<eddZeeed<eddZeed<eddZee ed<dS) TrainWorkerz"Metadata about a Ray Train worker.z6The global rank of the worker in the training cluster.r world_rankz)The local rank of the worker on its node. local_rankz-The rank of the worker's node in the cluster. node_rankz$The unique ID of the worker's actor.actor_idz6The unique ID of the node where the worker is running.node_idz7The IP address of the node where the worker is running.node_ipzThe process ID of the worker.pidz*A list of GPU IDs allocated to the worker.gpu_idsz'The current status of the worker actor.statusz-The resources allocated to this Train worker.r!z.The path to the log file for the Train worker. log_file_pathN)rrrrrr&intr$r'r(r)r"r*r+r,r-rr.rrr!rr/rrrrr%Ks2 r%c@sbeZdZUdZeddZeed<eddZeed<eddZ e eed<ed dZ e eed <d S) MemoryInfoz'Memory usage information for a process.z2The resident set size (RSS) memory usage in bytes.rrssz-The virtual memory size (VMS) usage in bytes.vmszThe number of page faults.pfaultszThe number of page-ins.pageinsN) rrrrrr2r0r$r3r4rr5rrrrr1hs r1c@sPeZdZUdZeddZeed<eddZe e e ed<eddZ e ed<d S) ProcessStatsz(CPU and memory statistics for a process.zThe percentage of CPU usage.r cpuPercentzOMemory statistics, including total memory, free memory, and memory usage ratio.memz"Detailed memory usage information. memoryInfoN)rrrrrr7r#r$r8rrr0r9r1rrrrr6rs r6c@s6eZdZUdZeddZeed<eddZeed<dS)ProcessGPUUsagez#GPU usage statistics for a process.zThe process ID.rr,zThe GPU memory usage in bytes.gpuMemoryUsageN) rrrrrr,r0r$r;rrrrr:~s r:c@seZdZUdZeddZeed<eddZe ed<eddZ eed<ed dZ e e ed <ed dZe ed <ed dZe ed<eddZeed<dS)GPUStatszStatistics for a GPU.z!The unique identifier of the GPU.ruuidzThe index of the GPU.indexzThe name of the GPU.namez&The percentage utilization of the GPU.utilizationGpuz'The amount of GPU memory used in bytes. memoryUsedz(The total amount of GPU memory in bytes. memoryTotalz0GPU usage statistics for the associated process. processInfoN)rrrrrr=r"r$r>r0r?r@rr#rArBrCr:rrrrr<s r<c@sBeZdZUdZedddZeeed<ee ddZ e e ed<dS) DecoratedTrainWorkerzJDetailed metadata for a Ray Train worker, including process and GPU stats.Nz1CPU and memory statistics for the worker process.r processStatszDA list of GPUs used by the worker process, with detailed statistics.)default_factoryr gpus) rrrrrrErr6r$listrGrr<rrrrrDs rDc@seZdZUdZeddZeed<eddZeed<eddZ e ed<ed dZ e eed <ed dZ eed <ed dZe eed<eddZeeed<eddZeeed<dS)TrainRunAttemptz:Metadata for an individual attempt to execute a Train run.z+Unique identifier for the parent Train run.rrun_idz6Unique identifier for this specific Train run attempt. attempt_idz6The current execution status of the Train run attempt.r.zLAdditional details about the status, including error messages if applicable. status_detailzGThe UNIX timestamp (in nanoseconds) when the Train run attempt started. start_time_nszlThe UNIX timestamp (in nanoseconds) when the Train run attempt ended. If null, the attempt is still ongoing. end_time_nszBThe resources (e.g., CPU, GPU) allocated to the Train run attempt.r!zLList of Train workers participating in this attempt, sorted by global ranks.workersN)rrrrrrJr"r$rKr.rrLrrMr0rNr!rrrOr%rrrrrIs0 rIc@(eZdZUdZeddZeeed<dS)DecoratedTrainRunAttemptzKDetailed metadata for a Train run attempt, including decorated worker data.zIA list of Train workers with detailed statistics, sorted by global ranks.rrON) rrrrrrOrrDr$rrrrrQ rQc@seZdZUdZeddZeed<eddZeed<eddZ eed<ed dZ eed <ed dZ e ed <ed dZ eeed<eddZeed<eddZeeed<eddZeeed<dS)TrainRunz?Metadata for a Ray Train run, including its details and status.z$Unique identifier for the Train run.ridz.Human-readable name assigned to the Train run.r?z.The Ray Job ID associated with this Train run.job_idz.Unique ID of the actor managing the Train run.controller_actor_idz.The current execution status of the Train run.r.zTAdditional details about the current status, including error messages if applicable.rLz?The UNIX timestamp (in nanoseconds) when the Train run started.rMzdThe UNIX timestamp (in nanoseconds) when the Train run ended. If null, the run is still in progress.rNz6The path to the log file for the Train run controller.controller_log_file_pathN)rrrrrrTr"r$r?rUrVr.r rLrrMr0rNrWrrrrrSs. rSc@s@eZdZUdZeddZeeed<edddZ e e ed<dS)DecoratedTrainRunzJDetailed metadata for a Ray Train run, including attempts and job details.z1A list of attempts made to execute the Train run.rattemptsNzADetailed information about the job that initiated this Train run. job_details) rrrrrrYrrQr$rZrrrrrrrXs rXc@rP)TrainRunsResponsez3Response containing a list of decorated Train runs.z,A list of Train runs with detailed metadata.r train_runsN) rrrrrr\rrXr$rrrrr[rRr[N)enumrtypingrrrray._common.pydantic_compatrr)ray.dashboard.modules.job.pydantic_modelsrray.util.annotationsr MAX_ERROR_STACK_TRACE_LENGTHr"r rrrr%r1r6r:r<rDrIrQrSrXr[rrrrsF