o nin@sbdZddlZddlZddlmZddlmZmZmZmZm Z m Z m Z ddl m Z mZddlZddlmZddlmZddlmZdd lmZdd lmZerVdd lmZmZd Zd ZGdddeZGdddZ GdddeZ!GdddZ"GdddeZ#GdddeZ$GdddeZ% d(ddde&d e&fd!d"Z'd#e&d e(fd$d%Z)d#e&d e(fd&d'Z*dS))zPublic API: artifacts.N)copy) TYPE_CHECKINGAnyListMappingOptionalSequenceUnion)Clientgql)public)normalize_exceptions) Paginator)termlog) deprecate)RetryingClientRuna fragment ArtifactTypesFragment on ArtifactTypeConnection { edges { node { id name description createdAt } cursor } pageInfo { endCursor hasNextPage } } a fragment ArtifactFilesFragment on Artifact { files(names: $fileNames, after: $fileCursor, first: $fileLimit) { edges { node { id name: displayName url sizeBytes storagePath mimetype updatedAt digest md5 directUrl } cursor } pageInfo { endCursor hasNextPage } } }c sxeZdZedeZ ddededede e ffdd Z e d d Z e d d Ze d dZddZddZZS) ArtifactTypesan query ProjectArtifacts( $entityName: String!, $projectName: String!, $cursor: String, ) {{ project(name: $projectName, entityName: $entityName) {{ artifactTypes(after: $cursor) {{ ...ArtifactTypesFragment }} }} }} {} 2cliententityprojectper_pagecs*||_||_||d}t|||dS)N) entityName projectName)rrsuper__init__)selfrrrrvariable_values __class__Y/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/wandb/apis/public/artifacts.pyrSs zArtifactTypes.__init__cCsdSNr!rr!r!r"lengthdszArtifactTypes.lengthcCs |jr|jddddSdS)Nr artifactTypespageInfo hasNextPageT last_responser$r!r!r"morei zArtifactTypes.morecC$|jr|jdddddSdS)Nrr&edgescursorr)r$r!r!r"r0rszArtifactTypes.cursorcC|jd|jidSNr0 variablesupdater0r$r!r!r"update_variablesyzArtifactTypes.update_variablescs2jddur gSfddjdddDS)Nrc s.g|]}tjjj|dd|dqSnodename) ArtifactTyperrr.0rr$r!r" s z1ArtifactTypes.convert_objects..r&r.r)r$r!r$r"convert_objects|s  zArtifactTypes.convert_objectsr)__name__ __module__ __qualname__r formatARTIFACTS_TYPES_FRAGMENTQUERYr strrintrpropertyr%r+r0r6r@ __classcell__r!r!rr"rAs.    rc@sxeZdZ ddededededeeeeff ddZd d Z e d d Z e d dZ e dddZddZddZdS)r;Nrrr type_nameattrscCs8||_||_||_||_||_|jdur|dSdSr#)rrrtype_attrsload)rrrrrLrMr!r!r"rs  zArtifactType.__init__cCsrtd}|jj||j|j|jdd}|dus'|ddus'|dddur/td|j|dd|_ |j S)Na query ProjectArtifactType( $entityName: String!, $projectName: String!, $artifactTypeName: String! ) { project(name: $projectName, entityName: $entityName) { artifactType(name: $artifactTypeName) { id name description createdAt } } } rrartifactTypeNamerr artifactTypeCould not find artifact type {}) r rexecuterrrNget ValueErrorrErO)rqueryresponser!r!r"rPs  zArtifactType.loadcC |jdSNidrOr$r!r!r"r] zArtifactType.idcCr[)Nr:r^r$r!r!r"r:r_zArtifactType.namercCst|j|j|j|jS)zArtifact collections.)ArtifactCollectionsrrrrNrrr!r!r" collectionsszArtifactType.collectionscCst|j|j|j||jSr#)ArtifactCollectionrrrrNrr:r!r!r" collectionszArtifactType.collectioncCsd|jdS)Nz)rNr$r!r!r"__repr__szArtifactType.__repr__r#rA)rBrCrDr rHrrrrrPrJr]r:r rbrergr!r!r!r"r;s, $    r;c sneZdZ ddededededeef fdd Zed d Z ed d Z ed dZ ddZ ddZ ZS)r`rrrrrLrcsJ||_||_||_|||d}tdtt||_t |||dS)NrQa% query ProjectArtifactCollections( $entityName: String!, $projectName: String!, $artifactTypeName: String! $cursor: String, ) {{ project(name: $projectName, entityName: $entityName) {{ artifactType(name: $artifactTypeName) {{ artifactCollections: {}(after: $cursor) {{ pageInfo {{ endCursor hasNextPage }} totalCount edges {{ node {{ id name description createdAt }} cursor }} }} }} }} }} ) rrrLr rE$artifact_collection_plural_edge_name.server_supports_artifact_collections_gql_edgesrGrr)rrrrrLrrrr!r"rs $zArtifactCollections.__init__cCs |jr|jddddSdS)NrrTartifactCollections totalCountr)r$r!r!r"r% r,zArtifactCollections.lengthcCs$|jr|jdddddSdS)NrrTrjr'r(Tr)r$r!r!r"r+zArtifactCollections.morecCs(|jr|jddddddSdS)NrrTrjr.r/r0r)r$r!r!r"r0zArtifactCollections.cursorcCr1r2r3r$r!r!r"r6&r7z$ArtifactCollections.update_variablescs$fddjddddDS)Ncs,g|]}tjjj|ddjqSr8)rcrrrrLr<r$r!r"r?*s z7ArtifactCollections.convert_objects..rrTrjr.r)r$r!r$r"r@)s  z#ArtifactCollections.convert_objectsrA)rBrCrDr rHrrIrrJr%r+r0r6r@rKr!r!rr"r`s*8   r`c@sreZdZ d7dedededededeeeeff dd Ze d d Z e d8d dZ e ddZ ddZdeddfddZdefddZe ddZe ddZejdeeddfddZe dd Zejd!eeddfd"d Ze d#d$Zejdeeddfd%d$Ze d&d'Zejdeeddfd(d'Zd)d*Zd+d,Zd-d.Zd/d0Zd1d2Zd9d3d4Zd5d6ZdS):rcNrrrr:rNrMcCs||_||_||_||_||_||_||_||_|dd|jddD|_ |jd|_ dd|jddD|_ t |j |_ dS)NcSg|]}|ddqS)r9aliasr!r=ar!r!r"r?Kz/ArtifactCollection.__init__..aliasesr. descriptioncSrnr8r!rpr!r!r"r?Mrrtags)rrr_name _saved_name_type _saved_typerOrP_aliases _description_tagsr _saved_tags)rrrrr:rNrMr!r!r"r9s  zArtifactCollection.__init__cCr[r\r^r$r!r!r"r]Pr_zArtifactCollection.idrcCst|j|j|j|j|j|dS)z Artifacts.)r) Artifactsrrrrwryrar!r!r" artifactsTszArtifactCollection.artifactscC|jS)zArtifact Collection Aliases.)rzr$r!r!r"rs`zArtifactCollection.aliasescCstdtt|j}|jj||j|j|j|j dd}|dus<| ddus<|d ddus<|dd ddurDt d|j|ddd}|duoU|d d k|_ |j dure|ddd|_ |j S) Na query ArtifactCollection( $entityName: String!, $projectName: String!, $artifactTypeName: String!, $artifactCollectionName: String!, $cursor: String, $perPage: Int = 1000 ) {{ project(name: $projectName, entityName: $entityName) {{ artifactType(name: $artifactTypeName) {{ artifactCollection: {}(name: $artifactCollectionName) {{ id name description createdAt tags {{ edges {{ node {{ id name }} }} }} aliases(after: $cursor, first: $perPage){{ edges {{ node {{ alias }} cursor }} pageInfo {{ endCursor hasNextPage }} }} }} artifactSequence(name: $artifactCollectionName) {{ __typename }} }} }} }} )rrrRartifactCollectionNamerSrrTartifactCollectionrUartifactSequence __typenameArtifactSequence)r rEartifact_collection_edge_namerirrVrrryrwrWrX _is_sequencerO)rrYrZsequencer!r!r"rPes6+2  zArtifactCollection.loadnew_typereturncCsptjtjjdd|stdtd|jd|d}|j|d}t|}|j j ||d||_||_ d S) z5Deprecated, change type directly with `save` instead.zZArtifactCollection.change_type(type) is deprecated, use ArtifactCollection.save() instead.) field_namewarning_messagez*Artifact collection needs to be a sequencez%Changing artifact collection type of z to ^ mutation MoveArtifactCollection( $artifactSequenceID: ID! $destinationArtifactTypeName: String! ) { moveArtifactSequence( input: { artifactSequenceID: $artifactSequenceID destinationArtifactTypeName: $destinationArtifactTypeName } ) { artifactCollection { id name description __typename } } } artifactSequenceIDdestinationArtifactTypeNamerSN) r Deprecated artifact_collection__change_type is_sequencerXrryr]r rrVrx)rrtemplatermutationr!r!r" change_types" zArtifactCollection.change_typecCr)z5Return whether the artifact collection is a sequence.)rr$r!r!r"rszArtifactCollection.is_sequencecCs4|r td}ntd}|jj|d|jiddS)z&Delete the entire artifact collection.a[ mutation deleteArtifactSequence($id: ID!) { deleteArtifactSequence(input: { artifactSequenceID: $id }) { artifactCollection { state } } } a^ mutation deleteArtifactPortfolio($id: ID!) { deleteArtifactPortfolio(input: { artifactPortfolioID: $id }) { artifactCollection { state } } } r]rSN)rr rrVr])rrr!r!r"deletes zArtifactCollection.deletecCr)z)A description of the artifact collection.r{r$r!r!r"rtrzArtifactCollection.descriptionrtcC ||_dSr#r)rrtr!r!r"rtr_cCr)z1The tags associated with the artifact collection.)r|r$r!r!r"rurzArtifactCollection.tagsrucCs$tdd|Dr td||_dS)Ncss|] }td| VqdS)z^[-\w]+([ ]+[-\w]+)*$N)rematchr=tagr!r!r" sz*ArtifactCollection.tags..z\Tags must only contain alphanumeric characters or underscores separated by spaces or hyphens)anyrXr|)rrur!r!r"ru s  cCr)z$The name of the artifact collection.rvr$r!r!r"r:rzArtifactCollection.namecCrr#rrdr!r!r"r:r_cCr)z$The type of the artifact collection.)rxr$r!r!r"rNrzArtifactCollection.typecCs|std||_dS)NzBType can only be changed if the artifact collection is a sequence.)rrXrx)rrNr!r!r"rN#s  cC6td}|j|j|jd}|jj||d|j|_dS)NaL mutation UpdateArtifactCollection( $artifactSequenceID: ID! $name: String $description: String ) { updateArtifactSequence( input: { artifactSequenceID: $artifactSequenceID name: $name description: $description } ) { artifactCollection { id name description } } } )rr:rtrSr r]rvrtrrVrwrrrr!r!r"_update_collection+s z%ArtifactCollection._update_collectioncCs2td}|j|jd}|jj||d|j|_dS)NrrrS)r r]rxrrVry)r type_mutationrr!r!r"_update_collection_typeJs  z*ArtifactCollection._update_collection_typecCr)NaO mutation UpdateArtifactPortfolio( $artifactPortfolioID: ID! $name: String $description: String ) { updateArtifactPortfolio( input: { artifactPortfolioID: $artifactPortfolioID name: $name description: $description } ) { artifactCollection { id name description } } } )artifactPortfolioIDr:rtrSrrr!r!r"_update_portfoliogs z$ArtifactCollection._update_portfoliocC6td}|jj||j|j|jdd|DdddS)Na mutation CreateArtifactCollectionTagAssignments( $entityName: String! $projectName: String! $artifactCollectionName: String! $tags: [TagInput!]! ) { createArtifactCollectionTagAssignments( input: { entityName: $entityName projectName: $projectName artifactCollectionName: $artifactCollectionName tags: $tags } ) { tags { id name tagCategoryName } } } cSg|]}d|iqStagNamer!rr!r!r"r? z0ArtifactCollection._add_tags..rrrrurSr rrVrrrw)r tags_to_add add_mutationr!r!r" _add_tagss zArtifactCollection._add_tagscCr)Nag mutation DeleteArtifactCollectionTagAssignments( $entityName: String! $projectName: String! $artifactCollectionName: String! $tags: [TagInput!]! ) { deleteArtifactCollectionTagAssignments( input: { entityName: $entityName projectName: $projectName artifactCollectionName: $artifactCollectionName tags: $tags } ) { success } } cSrrr!rr!r!r"r?rz3ArtifactCollection._delete_tags..rrSr)rtags_to_deletedelete_mutationr!r!r" _delete_tagss zArtifactCollection._delete_tagscCs|r||j|jkr|n|t|jt|j}t|jt|j}t |dkr6| |t |dkrA| |t |j|_dS)z4Persist any changes made to the artifact collection.rN) rrryrxrrsetr|r}lenrrr)rrrr!r!r"saves     zArtifactCollection.savecCsd|jd|jdS)Nz)rvrxr$r!r!r"rgr7zArtifactCollection.__repr__r#rA)rN) rBrCrDr rHrrrrrJr]r rrsrPrboolrrrtsetterrurr:rNrrrrrrrgr!r!r!r"rc8sb    L)     ) % rccseZdZdZ    ddededededed eeeefd eed e d ee ee efffd d Z e ddZe ddZe ddZddZZS)r~zAn iterable collection of artifact versions associated with a project and optional filter. This is generally used indirectly via the `Api`.artifact_versions method. Nrrrrcollection_namerNfiltersorderrruc sddlm} ||_||_||_||_|durddin||_t| tr%| gn| |_ ||_ |j|j|j |j|jt |jd} t dtt|| |_t|| |dS)Nr_gql_artifact_fragmentstate COMMITTED)rrrrNrera query Artifacts($project: String!, $entity: String!, $type: String!, $collection: String!, $cursor: String, $perPage: Int = 50, $order: String, $filters: JSONString) {{ project(name: $project, entityName: $entity) {{ artifactType(name: $type) {{ artifactCollection: {}(name: $collection) {{ name artifacts(filters: $filters, after: $cursor, first: $perPage, order: $order) {{ totalCount edges {{ node {{ ...ArtifactFragment }} version cursor }} pageInfo {{ endCursor hasNextPage }} }} }} }} }} }} {} )wandb.sdk.artifacts.artifactrrrrNrr isinstancerHrurjsondumpsr rErrirGrr) rrrrrrNrrrrurr4rr!r"rs2  !zArtifacts.__init__cCr-)NrrTrrrkr)r$r!r!r"r%/rlzArtifacts.lengthcC(|jr|jddddddSdS)NrrTrrr'r(Tr)r$r!r!r"r+8rmzArtifacts.morecC,|jr|jdddddddSdS)NrrTrrr.r/r0r)r$r!r!r"r0AszArtifacts.cursorcsXjddd}|didg}fdd|D}tjp!gfdd |DS) NrrTrrr.c3s<|]}tjjjjd|d|djVqdS):versionr9N)wandbArtifact _from_attrsrrrrrpr$r!r"rMs z,Artifacts.convert_objects..csg|] }|jr|qSr!)issubsetru)r=artifact) required_tagsr!r"r?Xs  z-Artifacts.convert_objects..)r*rWrru)rreartifact_edgesrr!)rrr"r@Js  zArtifacts.convert_objects)NNrN)rBrCrD__doc__r rHrrrrIr rrrJr%r+r0r@rKr!r!rr"r~s@   @   r~cs^eZdZ ddedddeeffdd Zed d Zed d Z ed dZ ddZ Z S) RunArtifactsloggedrrrunrrc sddlm}td|}td|}||_|dkr"d|_||_n|dkr-d|_||_ntd |j|j|j d }t |||dS) Nrra[ query RunOutputArtifacts( $entity: String!, $project: String!, $runName: String!, $cursor: String, $perPage: Int, ) { project(name: $project, entityName: $entity) { run(name: $runName) { outputArtifacts(after: $cursor, first: $perPage) { totalCount edges { node { ...ArtifactFragment } cursor } pageInfo { endCursor hasNextPage } } } } } aY query RunInputArtifacts( $entity: String!, $project: String!, $runName: String!, $cursor: String, $perPage: Int, ) { project(name: $project, entityName: $entity) { run(name: $runName) { inputArtifacts(after: $cursor, first: $perPage) { totalCount edges { node { ...ArtifactFragment } cursor } pageInfo { endCursor hasNextPage } } } } } routputArtifactsusedinputArtifactszmode must be logged or used)rrrunName) rrr rrun_keyrGrXrrr]rr) rrrmoderr output_query input_queryrrr!r"r^s0 zRunArtifacts.__init__cCs"|jr|jdd|jdSdS)Nrrrkr*rr$r!r!r"r%szRunArtifacts.lengthcCs&|jr|jdd|jddSdS)Nrrr'r(Trr$r!r!r"r+s zRunArtifacts.morecCs*|jr|jdd|jdddSdS)Nrrr.r/r0rr$r!r!r"r0s zRunArtifacts.cursorcs&fddjddjdDS)Nc sbg|]-}tj|dddd|ddddd|ddd|dd|djqS)r9rrrr:z{}:v{} versionIndex)rrrrErr<r$r!r"r?s z0RunArtifacts.convert_objects..rrr.rr$r!r$r"r@s  zRunArtifacts.convert_objects)rr) rBrCrDr rrIrrJr%r+r0r@rKr!r!rr"r]s M   rc seZdZedeZ  ddedddee e de ffd d Z e d d Ze d dZe ddZe ddZddZddZddZZS) ArtifactFilesaV query ArtifactFiles( $entityName: String!, $projectName: String!, $artifactTypeName: String!, $artifactName: String! $fileNames: [String!], $fileCursor: String, $fileLimit: Int = 50 ) {{ project(name: $projectName, entityName: $entityName) {{ artifactType(name: $artifactTypeName) {{ artifact(name: $artifactName) {{ ...ArtifactFilesFragment }} }} }} }} {} Nrrrzwandb.ArtifactnamesrcsV||_|j|j|j|j|d}|ds!t|jjj j dd|_t |||dS)N)rrrR artifactName fileNamesz0.12.21z storagePath )r source_entitysource_projectrN source_nameversion_supportedr rGlocsourcebodyreplacerr)rrrrrr4rr!r"rs zArtifactFiles.__init__cCs|jj|jj|jjgSr#)rrrr:r$r!r!r"pathszArtifactFiles.pathcCs|jjSr#)r file_countr$r!r!r"r%szArtifactFiles.lengthcCr)NrrTrfilesr'r(Tr)r$r!r!r"r+szArtifactFiles.morecCr)NrrTrrr.r/r0r)r$r!r!r"r0szArtifactFiles.cursorcCs|j|j|jddS)N) fileLimit fileCursor)r4r5rr0r$r!r!r"r6szArtifactFiles.update_variablescs(fddjdddddDS)Ncsg|] }tj|dqS)r9)r Filerr<r$r!r"r?sz1ArtifactFiles.convert_objects..rrTrrr.r)r$r!r$r"r@s zArtifactFiles.convert_objectscCsdd|jt|S)Nz/)rEjoinrrr$r!r!r"rg%szArtifactFiles.__repr__)Nr)rBrCrDr rEARTIFACT_FILES_FRAGMENTrGr rrrHrIrrJrr%r+r0r6r@rgrKr!r!rr"rs6     rFrrwarnrcCs |d}|s|rtd|S)Nz0.12.11zW&B Local Server version does not support ArtifactCollection gql edges; falling back to using legacy ArtifactSequence. Please update server to at least version 0.9.50.)rrtermwarn)rr supportedr!r!r"ri)s ri$server_supports_artifact_collectionscC |rdSdS)Nrrr!rr!r!r"r8s rcCr)NrjartifactSequencesr!rr!r!r"rh@s rh)F)+rrrrtypingrrrrrrr wandb_gqlr r r wandb.apisr wandb.apis.normalizer wandb.apis.paginatorrwandb.errors.termr wandb.sdk.librwandb.apis.publicrrrFrrr;r`rcr~rrrrirHrrhr!r!r!r"sP $     FKf4tvW