Vh ddlZddlmZmZmZddlmZddlmZm Z gdZ e e e e edfZGdde eZeGdd ZGd d e eZeGd d Zy)N)asdict dataclassfield)Enum)OptionalUnion) EventSourceEvent NodeState RdzvEventceZdZdZdZdZy)r z)Known identifiers of the event producers.AGENTWORKERN)__name__ __module__ __qualname____doc__rrT/home/dcms/DCMS/lib/python3.12/site-packages/torch/distributed/elastic/events/api.pyr r s3 E Frr ceZdZUdZeed<eed<dZeed<e e Z e ee fed<dZ ed eedfd dfd Zd efd Zy )r a} The class represents the generic event that occurs during the torchelastic job execution. The event can be any kind of meaningful action. Args: name: event name. source: the event producer, e.g. agent or worker timestamp: timestamp in milliseconds when event occurred. metadata: additional data that is associated with the event. namesourcer timestamp)default_factorymetadatac"|jSN serializeselfs r__str__z Event.__str__/~~rdatareturnct|tr|St|trtj|}t d|d<tdi|S)Nrr) isinstancer strjsonloadsr r% data_dicts r deserializezEvent.deserialize2sK dE "K dC  4(I))H*=> (!y!!rc>tjt|Srr*dumpsrr!s rr zEvent.serialize;zz&,''rN)rrrrr)__annotations__r rintrdictrEventMetadataValuer# staticmethodrr.r rrrr r st  I Is.3D.IHd3**+I "%W -"'""(3(rr c eZdZdZdZdZdZdZy)r z,The states that a node can be in rendezvous.INITRUNNING SUCCEEDEDFAILEDN)rrrrr9r:r;r<rrrr r ?s6 DGI Frr ceZdZUdZeed<eed<eed<eed<eed<eed<dZeed <d Z e eed <d Z e eed <dZ eed <dZ edeedfddfdZdefdZy )r a Dataclass to represent any rendezvous event. Args: name: Event name. (E.g. Current action being performed) run_id: The run id of the rendezvous message: The message describing the event hostname: Hostname of the node pid: The process id of the node node_state: The state of the node (INIT, RUNNING, SUCCEEDED, FAILED) master_endpoint: The master endpoint for the rendezvous store, if known rank: The rank of the node, if known local_id: The local_id of the node, if defined in dynamic_rendezvous.py error_trace: Error stack trace, if this is an error event. rrun_idmessagehostnamepid node_statemaster_endpointNranklocal_id error_tracec"|jSrrr!s rr#zRdzvEvent.__str__er$rr%r&ct|tr|St|trtj|}t d|d<tdi|S)NrBr)r(r r)r*r+r r,s rr.zRdzvEvent.deserializehsK dI &K dC  4(I"+Il,C"D ,%9%%rc>tjt|Srr0r!s rr zRdzvEvent.serializeqr2r)rrrrr)r3r4r rDrErrFrGr#r7rr.r rrrr r Hs I K LM HOSD(3-"Hhsm"K &%[ 01&k&&(3(rr )r* dataclassesrrrenumrtypingrr__all__r)r4floatboolr6r r r r rrrrQs 00" =3UD$67#t (( (DT )()( )(r