o hd @sddlZddlmZmZmZddlmZddlmZm Z m Z gdZ e e e eedfZGddde eZeGdd d ZGd d d e eZeGd d d ZdS)N)asdict dataclassfield)Enum)DictUnionOptional) EventSourceEvent NodeState RdzvEventc@seZdZdZdZdZdS)r z3 Known identifiers of the event producers. AGENTWORKERN)__name__ __module__ __qualname____doc__r rrrZ/var/www/html/ai/venv/lib/python3.10/site-packages/torch/distributed/elastic/events/api.pyr sr c@s|eZdZUdZeed<eed<dZeed<e e dZ e ee fed<dd Zed eedfd dfd d Zd efddZdS)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_factorymetadatacC|SN serializeselfrrr__str__.z Event.__str__datareturncC@t|tr|St|trt|}t|d|d<tdi|S)Nrr) isinstancer strjsonloadsr r" data_dictrrr deserialize1   zEvent.deserializecCtt|Srr'dumpsrrrrrr:zEvent.serializeN)rrrrr&__annotations__r rintrdictrrEventMetadataValuer staticmethodrr+rrrrrr s   r c@s eZdZdZdZdZdZdZdS)r z6 The states that a node can be in rendezvous. INITRUNNING SUCCEEDEDFAILEDN)rrrrr6r7r8r9rrrrr >s r c@seZdZUdZeed<eed<eed<eed<eed<eed<dZeed <d Z e eed <d Z e eed <dZ eed <ddZ edeedfddfddZdefddZd 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_tracecCrrrrrrrr fr!zRdzvEvent.__str__r"r#cCr$)Nr>r)r%r r&r'r(r r)rrrr+ir,zRdzvEvent.deserializecCr-rr.rrrrrrr0zRdzvEvent.serialize)rrrrr&r1r2r r@rArrBrCr r5rr+rrrrrr Is   r )r' dataclassesrrrenumrtypingrrr__all__r&r2floatboolr4r r r r rrrrs  !