o wOiO @sddlZddlmZmZmZddlmZddlmZm Z m Z e e e e edfZGddde eZeGdddZeGd d d ZdS) N)asdict dataclassfield)Enum)DictOptionalUnionc@seZdZdZdZdZdS) EventSourcez3 Known identifiers of the event producers. AGENTWORKERN)__name__ __module__ __qualname____doc__r r rrK/home/ubuntu/.local/lib/python3.10/site-packages/torchelastic/events/api.pyr sr c@s|eZdZUdZeed<eed<dZeed<e e dZ e ee fed<dd Zed eedfd dfd d Zd efddZdS)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_factorymetadatacC|SN serializeselfrrr__str__-z Event.__str__datareturncCs@t|tr|St|trt|}t|d|d<tdi|S)Nrr) isinstancerstrjsonloadsr r data_dictrrr deserialize0s   zEvent.deserializecCtt|Srr$dumpsrrrrrr9zEvent.serializeN)r r rrr#__annotations__r rintrdictrrEventMetadataValuer staticmethodrr(rrrrrrs   rc@seZdZUdZeed<eed<eed<eed<eed<dZeeed<dZeeed <dZ eeed <d d Z e d e edfddfddZ defddZdS)TsmEventa The class represents the event produced by tsm api calls. Arguments: session: Session id that was used to execute request. scheduler: Scheduler that is used to execute request api: Api name unix_user: Unix user that executes request source_hostname: Hostname of server that executes request app_id: Unique id that is set by the underlying scheduler runcfg: Run config that was used to schedule app. session schedulerapi unix_usersource_hostnameNapp_idruncfg raw_exceptioncCrrrrrrrrUrzTsmEvent.__str__r r!cCs0t|tr|St|trt|}tdi|S)Nr)r"r2r#r$r%r&rrrr(Xs   zTsmEvent.deserializecCr)rr*rrrrrar,zTsmEvent.serialize)r r rrr#r-r8rr9r:rr1rr(rrrrrr2=s  r2)r$ dataclassesrrrenumrtypingrrrr#r.floatboolr0r rr2rrrrs  !