U d& @sddlZddlmZmZmZddlmZddlmZm Z m Z e e e e edfZGddde eZeGdddZGd d d e eZeGd d d ZdS) N)asdict dataclassfield)Enum)DictUnionOptionalc@seZdZdZdZdZdS) EventSourcez3 Known identifiers of the event producers. AGENTWORKERN)__name__ __module__ __qualname____doc__r r rrH/tmp/pip-unpacked-wheel-ua33x9lu/torch/distributed/elastic/events/api.pyr sr c@szeZdZUdZeed<eed<dZeed<e e dZ e ee fed<dd Zeeedfdd d d Zed ddZdS)Eventa{ 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 occured. metadata: additional data that is associated with the event. namesourcer timestamp)default_factorymetadatacCs|SN serializeselfrrr__str__-sz Event.__str__datareturncCs<t|tr|St|tr"t|}t|d|d<tf|S)Nr) isinstancerstrjsonloadsr rZ data_dictrrr deserialize0s    zEvent.deserializer cCstt|Srr#dumpsrrrrrr9szEvent.serializeN)r r rrr"__annotations__r rintrdictrrEventMetadataValuer staticmethodrr&rrrrrrs   rc@s eZdZdZdZdZdZdZdS) NodeStatez6 The states that a node can be in rendezvous. INITRUNNING SUCCEEDEDFAILEDN)r r rrr0r1r2r3rrrrr/=s r/c@seZdZUdZeed<eed<eed<eed<eed<eed<dZeed <d Z e eed <d Z e eed <dZ eed <ddZ eeedfddddZedddZd S) RdzvEventa 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. rZrun_idmessagehostnamepid node_statemaster_endpointNranklocal_id error_tracecCs|SrrrrrrreszRdzvEvent.__str__rcCs<t|tr|St|tr"t|}t|d|d<tf|S)Nr8)r!r4r"r#r$r/r%rrrr&hs    zRdzvEvent.deserializer'cCstt|Srr(rrrrrqszRdzvEvent.serialize)r r rrr"r*r+r/r:r;rr<r=rr.rr&rrrrrr4Hs   r4)r#Z dataclassesrrrenumrtypingrrrr"r+floatboolr-r rr/r4rrrr s  !