a h @sddlZddlmZmZmZddlmZddlmZm Z gdZ e e e e edfZGddde eZeGdd d ZGd d d e eZeGd d d ZdS)N)asdict dataclassfield)Enum)OptionalUnion) EventSourceEvent NodeState RdzvEventc@seZdZdZdZdZdS)rz)Known identifiers of the event producers.AGENTWORKERN)__name__ __module__ __qualname____doc__r r rrR/var/www/auris/lib/python3.9/site-packages/torch/distributed/elastic/events/api.pyrsrc@szeZdZUdZeed<eed<dZeed<e e dZ e ee fed<dd Z eeedfdd d d Zed ddZdS)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_factorymetadatacCs|SN serializeselfrrr__str__/sz Event.__str__datareturncCs@t|tr|St|tr"t|}t|d|d<tfi|S)Nr) isinstancer strjsonloadsrr Z data_dictrrr deserialize2s    zEvent.deserializer!cCstt|Srr$dumpsrrrrrr;szEvent.serializeN)rrrrr#__annotations__rrintrdictrEventMetadataValuer staticmethodrr'rrrrrr s   r c@s eZdZdZdZdZdZdZdS)r z,The states that a node can be in rendezvous.INITRUNNING SUCCEEDEDFAILEDN)rrrrr0r1r2r3rrrrr ?s r c@seZdZUdZeed<eed<eed<eed<eed<eed<dZeed <d Z e eed <d Z e eed <dZ eed <ddZ eeedfddddZedddZd 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. rZrun_idmessagehostnamepid node_statemaster_endpointNranklocal_id error_tracecCs|SrrrrrrreszRdzvEvent.__str__rcCs@t|tr|St|tr"t|}t|d|d<tfi|S)Nr7)r"r r#r$r%r r&rrrr'hs    zRdzvEvent.deserializer(cCstt|Srr)rrrrrqszRdzvEvent.serialize)rrrrr#r+r,r r9r:rr;r<rr/rr'rrrrrr Hs   r )r$Z dataclassesrrrenumrtypingrr__all__r#r,floatboolr.rr r r rrrr s "