o wid6@sUdZddlmZddlmZmZmZmZmZm Z ddl m Z m Z ddl mZddlmZmZmZddlmZdd lmZmZdd lmZdd lmZmZmZmZdd lm Z m!Z!dd l"m#Z#ddl$m%Z%m&Z&m'Z'm(Z(m)Z)ddl*m+Z+m,Z,m-Z-ddl.m/Z/m0Z0m1Z1erddl2m3Z3Gddde&Z4GdddeZ5GdddeZ6GdddeZ7GdddeZ8GdddeZ9GdddeZ:Gd d!d!e#Z;Gd"d#d#eZGd(d)d)e=Z?Gd*d+d+e=Z@Gd,d-d-e<ZAGd.d/d/eAZBGd0d1d1eAZCee e>e?e@eBeCfe d2d3fZDeeDjEZFd4eGd5<Gd6d7d7ZHGd8d9d9ZIeJe9Je5Je>Je?Je@JeBJdgd:d;eFDd7d9d<d=d>ZKd?S)@z$Events that trigger W&B Automations.) annotations) TYPE_CHECKING AnnotatedAnyLiteralOptionalUnion)AfterValidatorField)get_args)GQLBasemodel_validatorpydantic_isinstancenameof)AndMongoLikeFilter)FilterableField)MetricChangeFilterMetricThresholdFilter MetricValMetricZScoreFilter) StateFilter StateOperand)FilterEventFields) JsonEncodedLenientStrEnum ensure_jsonwrap_mutation_event_filterwrap_run_event_run_filter) InputActionInputActionTypesSavedActionTypes)ArtifactCollectionScopeAutomationScope ProjectScope NewAutomationc@s0eZdZdZdZdZdZdZdZdZ dZ d Z d S) EventTypez.The type of event that triggers an automation.UPDATE_ARTIFACT_ALIASCREATE_ARTIFACTADD_ARTIFACT_ALIAS LINK_MODEL RUN_METRICRUN_METRIC_CHANGE RUN_STATERUN_METRIC_ZSCOREN) __name__ __module__ __qualname____doc__r*r+r, LINK_ARTIFACTRUN_METRIC_THRESHOLDr/r0r1r8r8L/home/ubuntu/.local/lib/python3.10/site-packages/wandb/automations/events.pyr)&sr)c@eZdZUeZded<dS)_WrappedSavedEventFilterzJsonEncoded[MongoLikeFilter]filterNr2r3r4rr<__annotations__r8r8r8r9r;As r;c@<eZdZUejZded<ded<edded d d Z d S)_WrappedMetricThresholdFilterzSAnnotated[Literal[EventType.RUN_METRIC_THRESHOLD], Field(exclude=True, repr=False)] event_typerthreshold_filterbeforemodevrreturncCt|tr ||dS|S)N)rB)rrclsrFr8r8r9_nest_inner_filterM  z0_WrappedMetricThresholdFilter._nest_inner_filterNrFrrGr) r2r3r4r)r7rAr>r classmethodrKr8r8r8r9r@E  r@c@r?)_WrappedMetricChangeFilterzPAnnotated[Literal[EventType.RUN_METRIC_CHANGE], Field(exclude=True, repr=False)]rAr change_filterrCrDrFrrGcCrH)N)rQ)rrrIr8r8r9rK^rLz-_WrappedMetricChangeFilter._nest_inner_filterNrM) r2r3r4r)r/rAr>r rNrKr8r8r8r9rPVrOrPc@r?)_WrappedMetricZScoreFilterzPAnnotated[Literal[EventType.RUN_METRIC_ZSCORE], Field(exclude=True, repr=False)]rAr zscore_filterrCrDrFrrGcCrH)N)rS)rrrIr8r8r9rKos  z-_WrappedMetricZScoreFilter._nest_inner_filterNrM) r2r3r4r)r1rAr>r rNrKr8r8r8r9rRgrOrRc@sNeZdZUeZded< ded< dZded< edd edd dZ dS)RunMetricFiltermAnnotated[JsonEncoded[MongoLikeFilter], AfterValidator(wrap_run_event_run_filter), Field(alias='run_filter')]runzAnnotated[Union[_WrappedMetricThresholdFilter, _WrappedMetricChangeFilter, _WrappedMetricZScoreFilter], Field(alias='run_metric_filter')]metricNzfAnnotated[Optional[JsonEncoded[MetricThresholdFilter]], Field(alias='metric_filter', deprecated=True)]legacy_metric_filterrCrDrFrrGcCst|tttfr ||dS|S)N)rW)rrrrrIr8r8r9_nest_metric_filters   z#RunMetricFilter._nest_metric_filterrM) r2r3r4rrVr>rXr rNrYr8r8r8r9rTws   rTc@sDeZdZUdZeZded< ded< eddedd d Z d S)RunStateFilterzIRepresents a filter for triggering events based on changes in run states.rUrVz7Annotated[StateFilter, Field(alias='run_state_filter')]staterCrDrFrrGcCrH)N)r[)rrrIr8r8r9_nest_state_filters  z!RunStateFilter._nest_state_filterNrM) r2r3r4r5rrVr>r rNr\r8r8r8r9rZs  rZc@s"eZdZUdZded<ded<dS) SavedEventz+A triggering event from a saved automation.z(Annotated[EventType, Field(frozen=True)]rAzMJsonEncoded[Union[_WrappedSavedEventFilter, RunMetricFilter, RunStateFilter]]r<N)r2r3r4r5r>r8r8r8r9r]s r]c@s<eZdZUded<ded< ded<dd d ZdddZdS)_BaseEventInputr)rAr%scopezJsonEncoded[Any]r<actionr!rGr(cCs<ddlm}t|ttfr|||dStdtt|)zFDefine a new Automation in which this event triggers the given action.rr')eventr`zExpected a valid action, got: ) automationsr( isinstancer"r# TypeErrorrtype)selfr`r(r8r8r9thens  z_BaseEventInput.thenothercCs ||S)z4Implement `event >> action` to define an automation.)rg)rfrhr8r8r9 __rshift__s z_BaseEventInput.__rshift__N)r`r!rGr()rhr!rGr()r2r3r4r>rgrir8r8r8r9r^s   r^c@r:)_BaseMutationEventInputzSAnnotated[JsonEncoded[MongoLikeFilter], AfterValidator(wrap_mutation_event_filter)]r<Nr=r8r8r8r9rjs  rjc@ eZdZUdZejZded<dS)OnLinkArtifactaA new artifact is linked to a collection. Examples: Define an event that triggers when an artifact is linked to the collection "my-collection" with the alias "prod": ```python from wandb import Api from wandb.automations import OnLinkArtifact, ArtifactEvent api = Api() collection = api.artifact_collection(name="my-collection", type_name="model") event = OnLinkArtifact( scope=collection, filter=ArtifactEvent.alias.eq("prod"), ) ``` z Literal[EventType.LINK_ARTIFACT]rAN)r2r3r4r5r)r6rAr>r8r8r8r9rl rlc@rk)OnAddArtifactAliasaA new alias is assigned to an artifact. Examples: Define an event that triggers whenever the alias "prod" is assigned to any artifact in the collection "my-collection": ```python from wandb import Api from wandb.automations import OnAddArtifactAlias, ArtifactEvent api = Api() collection = api.artifact_collection(name="my-collection", type_name="model") event = OnAddArtifactAlias( scope=collection, filter=ArtifactEvent.alias.eq("prod"), ) ``` z%Literal[EventType.ADD_ARTIFACT_ALIAS]rAN)r2r3r4r5r)r,rAr>r8r8r8r9rnrmrnc@(eZdZUdZejZded<ded<dS)OnCreateArtifactaA new artifact is created. Examples: Define an event that triggers when a new artifact is created in the collection "my-collection": ```python from wandb import Api from wandb.automations import OnCreateArtifact api = Api() collection = api.artifact_collection(name="my-collection", type_name="model") event = OnCreateArtifact(scope=collection) ``` z"Literal[EventType.CREATE_ARTIFACT]rAr$r_N)r2r3r4r5r)r+rAr>r8r8r8r9rps rpc@seZdZUded<dS)_BaseRunEventInputr&r_N)r2r3r4r>r8r8r8r9rq3s rqc@s<eZdZUdZded<ded< eddedd d Zd S) OnRunMetricaA run metric satisfies a user-defined condition. Examples: Define an event that triggers for any run in project "my-project" when the average of the last 5 values of metric "my-metric" exceeds 123.45: ```python from wandb import Api from wandb.automations import OnRunMetric, RunEvent api = Api() project = api.project(name="my-project") event = OnRunMetric( scope=project, filter=RunEvent.metric("my-metric").avg(5).gt(123.45), ) ``` zaLiteral[EventType.RUN_METRIC_THRESHOLD, EventType.RUN_METRIC_CHANGE, EventType.RUN_METRIC_ZSCORE]rAzJsonEncoded[RunMetricFilter]r<rCrDdatarrGcCs>t|tr|d}rtt|}i|d|jjiS|S)zInfer the event type from the inner filter during validation. This supports both "threshold" and "change" metric filters, which can only be determined after parsing and validating the inner JSON data. r<rA)rcdictgetrTmodel_validate_jsonrrWrA)rJrs raw_filter parsed_filterr8r8r9_infer_event_typeVszOnRunMetric._infer_event_typeN)rsrrGr)r2r3r4r5r>r rNryr8r8r8r9rr8s rrc@ro) OnRunStateaA run state changes. Examples: Define an event that triggers for any run in project "my-project" when its state changes to "finished" (i.e. succeeded) or "failed": ```python from wandb import Api from wandb.automations import OnRunState api = Api() project = api.project(name="my-project") event = OnRunState( scope=project, filter=RunEvent.state.in_(["finished", "failed"]), ) ``` zLiteral[EventType.RUN_STATE]rAzJsonEncoded[RunStateFilter]r<N)r2r3r4r5r)r0rAr>r8r8r8r9rzfs rzrA) discriminatorztuple[type, ...]InputEventTypesc@s*eZdZeddZeZed ddZd S) RunEvent display_name) server_namenamestrrGrcCs t|dS)z!Define a metric filter condition.r)rrr8r8r9rWs zRunEvent.metricN)rrrGr) r2r3r4rrrr[ staticmethodrWr8r8r8r9r}s  r}c@seZdZeZdS) ArtifactEventN)r2r3r4raliasr8r8r8r9rs rccs|]}t|VqdS)Nr).0rJr8r8r9 srrrrN)Lr5 __future__rtypingrrrrrrpydanticr r typing_extensionsr wandb._pydanticr r rwandb._strutilsr_filtersrr_filters.expressionsr_filters.run_metricsrrrr_filters.run_statesrr _generatedr _validatorsrrrrr actionsr!r"r#scopesr$r%r&rbr(r)r;r@rPrRrTrZr]r^rjrlrnrprqrrrz InputEvent __origin__r|r>r}r model_rebuild__all__r8r8r8r9s       (.