j' ddlZddlmZmZmZddlmZddlmZm Z m Z gdZ e e e eedfZGdde eZeGdd ZGd d e eZeGd d ZdS)N)asdict dataclassfield)Enum)DictOptionalUnion) EventSourceEvent NodeState RdzvEventceZdZdZdZdZdS)r z)Known identifiers of the event producers.AGENTWORKERN)__name__ __module__ __qualname____doc__rrUC:\Projects\Historical\venv\Lib\site-packages\torch/distributed/elastic/events/api.pyr r s33 E FFFrr ceZdZUdZeed<eed<dZeed<e e Z e ee fed<dZed eedfd dfd Zd efd Zd S)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*|SN serializeselfs r__str__z Event.__str__/~~rdatareturnct|tr|St|trtj|}t |d|d<tdi|S)Nrr) isinstancer strjsonloadsr r& data_dicts r deserializezEvent.deserialize2sb dE " " K dC  ) 4((I))H*=> (!!y!!!rcDtjt|Srr+dumpsrr"s rr!zEvent.serialize;z&,,'''rN)rrrrr*__annotations__r rintrdictrrEventMetadataValuer$ staticmethodr r/r!rrrr r s   III Is.3eD.I.I.IHd3**+III   "%W -"'"""\"(3((((((rr c"eZdZdZdZdZdZdZdS)r z,The states that a node can be in rendezvous.INITRUNNING SUCCEEDEDFAILEDN)rrrrr:r;r<r=rrrr r ?s(66 DGI FFFrr ceZdZUdZeed<eed<eed<eed<eed<eed<dZeed <d Z e eed <d Z e eed <dZ eed <dZ edeedfddfdZdefdZd S)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*|Srr r"s rr$zRdzvEvent.__str__er%rr&r'ct|tr|St|trtj|}t |d|d<tdi|S)NrCr)r)r r*r+r,r r-s rr/zRdzvEvent.deserializehsb dI & & K dC  ) 4((I"+Il,C"D ,%%9%%%rcDtjt|Srr1r"s rr!zRdzvEvent.serializeqr3r)rrrrr*r4r5r rErFrrGrHr$r8r r/r!rrrr r Hs III KKK LLLMMM HHHOSD(3-"Hhsm"""K   &%[ 01&k&&&\&(3((((((rr )r+ dataclassesrrrenumrtypingrrr __all__r*r5floatboolr7r r r r rrrrRsL 0000000000(((((((((( = < <3UD$67#t ((((((( (DT )()()()()()()( )()()(r