o niK@sJUdZddlmZddlZddlmZddlmZmZm Z m Z m Z m Z ddl mZmZerWddlmZmZmZmZddlmZe d ed Ze d Zeeeeeed ffZd Zded<e d1d2ddZe d1d3ddZeeffd4ddZd5ddZ e!d Z"d!ed"<d6d$d%Z#e d&e d efd Z$ d7d(d)Z%d7d*d+Z&d8d/d0Z'dS)9z=Internal validation utilities that are specific to artifacts.) annotationsN)wraps) TYPE_CHECKINGAnyCallableTypeVarcastoverload)ArtifactFinalizedErrorArtifactNotLoggedError) CollectionFinalIterableUnion)Artifact ArtifactT)boundT.zwandb-registry-z Final[str]REGISTRY_PREFIXobj Iterable[T] base_type ClassInforeturnlist[T]cCdSNrrrr]/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/wandb/sdk/artifacts/_validators.py always_listr cCrrrrrrrr r!rcCst||r|gSt|S)atReturn a guaranteed list of objects from a single instance OR iterable of such objects. By default, assume the returned list should have string-like elements (i.e. `str`/`bytes`). Adapted from `more_itertools.always_iterable`, but simplified for internal use. See: https://more-itertools.readthedocs.io/en/stable/api.html#more_itertools.always_iterable ) isinstancelistrrrrr "saliasesCollection[str] | str list[str]cs:t|}dtfdd|Drtdd|S)zValidate the artifact aliases and return them as a list. Raises: ValueError: If any of the aliases contain invalid characters. )/:c3s"|] }D]}||vVqqdSrr).0aliaschar invalid_charsrr 6s z#validate_aliases..z:Aliases must not contain any of the following characters: z, )r any ValueErrorjoin)r$ aliases_listrr,rvalidate_aliases-sr3z^[-\w]+( +[-\w]+)*$zre.Pattern[str]_VALID_TAG_PATTERNtagscCs0t|}tdd|Drtdtt|S)aValidate the artifact tag names and return them as a deduped list. In the case of duplicates, only keep the first tag, and otherwise maintain the order of appearance. Raises: ValueError: If any of the tags contain invalid characters. css|] }t| VqdSr)r4match)r)tagrrrr.Jsz validate_tags..zqInvalid tag(s). Tags must only contain alphanumeric characters separated by hyphens, underscores, and/or spaces.)r r/r0r#dictfromkeys)r5 tags_listrrr validate_tags@s r; DecoratedFmethodc(jtd fdd }tt|S) zDecorator to ensure that an Artifact method can only be called if the artifact has been logged. If the method is called on an artifact that's not logged, `ArtifactNotLoggedError` is raised. selfrargsrkwargsrcs*|r t|d|g|Ri|SN)fullnamer)is_draftr r?r@rAr=method_fullnamerrwrapper^s zensure_logged..wrapperNr?rr@rrArrr __qualname__rrr<r=rHrrFr ensure_loggedV rMcr>) zDecorator to ensure that an `Artifact` method can only be called if the artifact isn't finalized. If the method is called on an artifact that's not logged, `ArtifactFinalizedError` is raised. r?rr@rrArcs(|jr t|d|g|Ri|SrB)_finalr rErFrrrHos z%ensure_not_finalized..wrapperNrIrJrLrrFrensure_not_finalizedgrNrPprojectstrboolcCs |tSr) startswithr)rQrrris_artifact_registry_projectxs rU).)rrrrrr)rrrrrr)rrrrrr)r$r%rr&)r5r%rr&)r=r<rr<)rQrRrrS)(__doc__ __future__rre functoolsrtypingrrrrrr wandb.sdk.artifacts.exceptionsr r r r rrwandb.sdk.artifacts.artifactrrrtypetuplerr__annotations__r rRbytesr3compiler4r;r<rMrPrUrrrrs4