o -wi@sdZddlmZddlmZddlmZddlmZddl m Z ddl m Z ddl mZdd lmZdd lmZd ZGd d d edZGddde ZdS)zPublic API: projects.)suppress) HTTPError)gql)public)Attrs)normalize_exceptions) Paginator)RetryingClient)ipythonzdfragment ProjectFragment on Project { id name entityName createdAt isBenchmark }c szeZdZdZedeZ ddede de ddffdd Z e dd d Z e d dZe ddZddZddZZS)Projectsz,An iterable collection of `Project` objects.a query Projects($entity: String, $cursor: String, $perPage: Int = 50) {{ models(entityName: $entity, after: $cursor, first: $perPage) {{ edges {{ node {{ ...ProjectFragment }} cursor }} pageInfo {{ endCursor hasNextPage }} }} }} {} 2cliententityper_pagereturncs*||_||_d|ji}t|||dS)aAn iterable collection of `Project` objects. Args: client: The API client used to query W&B. entity: The entity which owns the projects. per_page: The number of projects to fetch per request to the API. rN)r rsuper__init__)selfr rr variables __class__W/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/wandb/apis/public/projects.pyr/s  zProjects.__init__NcCsdSNrrrrrlengthCszProjects.lengthcCs|jr |jdddSdS)NmodelspageInfo hasNextPageT last_responserrrrmoreHsz Projects.morecCs |jr|jddddSdS)Nredgescursorrrrrrr$OszProjects.cursorcsfddjddDS)Ncs*g|]}tjj|dd|dqSnodename)Projectr r).0prrr Wsz,Projects.convert_objects..rr"rrrrrconvert_objectsVs  zProjects.convert_objectscCsd|jdS)Nz )rrrrr__repr__\szProjects.__repr__r )rN)__name__ __module__ __qualname____doc__rformatPROJECT_FRAGMENTQUERYr strintrpropertyrr!r$r,r. __classcell__rrrrr s0   r r(c seZdZdZedZdededededdf fdd Z e d d Z e d d Z dddZ defddZddZedddZeddZedZe defddZZS) r(z"A project is a namespace for runs.z query Project($project: String!, $entity: String!) { project(name: $project, entityName: $entity) { id } } r rprojectattrsrcs&tt|||_||_||_dS)aA single project associated with an entity. Args: client: The API client used to query W&B. entity: The entity which owns the project. project: The name of the project to query. attrs: The attributes of the project. N)rrdictr r'r)rr rr;r<rrrrms zProject.__init__cCs |j|jgSr)rr'rrrrpaths z Project.pathcCs|jjd|jdgS)N/ workspace)r app_urljoinr>rrrrurlsz Project.urlFcCsF|jd}d|d}d}|r|d7}td}|d|d|d S) z;Generate HTML containing an iframe displaying this project.z ?jupyter=truezborder:none;width:100%;height:zpx;z display:none;r;z )rCr toggle_button)rheighthiddenrCstyleprefixrrrto_htmls   zProject.to_htmlcCs|Sr)rKrrrr _repr_html_szProject._repr_html_cCsdd|jS)Nz r?)r4rBr>rrrrr.szProject.__repr__r cCst|j|j|jSr)r ArtifactTypesr rr')rrrrrartifacts_typesszProject.artifacts_typescshtdtj}jjd}j||}|dr#|dddkr%gSfdd|ddd DS) Naw query GetSweeps($project: String!, $entity: String!) {{ project(name: $project, entityName: $entity) {{ totalSweeps sweeps {{ edges {{ node {{ ...SweepFragment }} cursor }} pageInfo {{ endCursor hasNextPage }} }} }} }} {} )r;rr; totalSweepsc s*g|]}tjjj|ddqSr%)rSweepr rr')r)errrr+s z"Project.sweeps..sweepsr") rr4rSWEEP_FRAGMENTr'rr executeget)rqueryvariable_valuesretrrrrSs zProject.sweepsz query ProjectID($projectName: String!, $entityName: String!) { project(name: $projectName, entityName: $entityName) { id } } c Cstt |jdWdS1swY|j|jd}z,|j|j|}|dr5|dds>t d|jd|dd|jd<|jdWSt tt fyc}zt d||d}~ww)Nid) projectName entityNamer;zProject z not foundzUnable to fetch project ID: ) r LookupError_attrsr'rr rU _PROJECT_IDrV ValueErrorr TypeError)rrXdatarRrrrrZs   z Project.id)rDFr/)r0r1r2r3rr6r r7r=rr9r>rCrKrLr.rrNrSr_rZr:rrrrr(`s@       ' N)r3 contextlibrrequestsr wandb_gqlr wandb.apisrwandb.apis.attrsrwandb.apis.normalizerwandb.apis.paginatorrwandb.apis.public.apir wandb.sdk.libr r5r r(rrrrs          H