o
ni @ st d Z ddlZddlmZ ddlmZ ddlZddlmZ ddlm Z ddl
m Z ddl m
Z
d ZG d
d d e ZdS ) zPublic API: sweeps. N)Optional)gql)util)public)Attrs)ipythonzfragment SweepFragment on Sweep {
id
name
method
state
description
displayName
bestLoss
config
createdAt
updatedAt
runCount
}
c s e Zd ZdZedZedZd' fdd Zedd Z ed d
Z
ed d Z d(de fddZ
edd Zd'ddZedee fddZedd Zedd Zed d Ze d)d d Zd*d!d"Zdefd#d$Zd%d& Z Z S )+Sweepa A set of runs associated with a sweep.
Examples:
Instantiate with:
```
api = wandb.Api()
sweep = api.sweep(path / to / sweep)
```
Attributes:
runs: (`Runs`) list of runs
id: (str) sweep id
project: (str) name of project
config: (str) dictionary of sweep configuration
state: (str) the state of the sweep
expected_run_count: (int) number of expected runs for the sweep
aT
query Sweep($project: String, $entity: String, $name: String!) {
project(name: $project, entityName: $entity) {
sweep(sweepName: $name) {
id
name
state
runCountExpected
bestLoss
config
}
}
}
a3
query Sweep($project: String, $entity: String, $name: String!) {
project(name: $project, entityName: $entity) {
sweep(sweepName: $name) {
id
name
state
bestLoss
config
}
}
}
Nc sD t t|pi || _|| _|| _|| _g | _| j| d d S )N)force) super__init__dictclient_entityprojectidrunsload)selfr
entityr sweep_idattrs __class__ V/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/wandb/apis/public/sweeps.pyr R s zSweep.__init__c C s | j S N)r r r r r r ] s z Sweep.entityc C s t d | jS )Nz>Sweep.username is deprecated. please use Sweep.entity instead.)wandbtermwarnr r r r r usernamea s
zSweep.usernamec C s t | jd S )Nconfig)r load_yaml_attrsr r r r r f s z Sweep.configFr c C sN |s| j s$| | j| j| j| j}|d u r td| |j | _ |j| _| j S )NzCould not find sweep {}) r" getr
r r r
ValueErrorformatr )r r sweepr r r r j s
z
Sweep.loadc C s\ | j dr*| jdr,| jd dd}|dkrdnd}tj|| jd d S d S d S )Nr metricgoalminimize+-name)r" r# r r QueryGeneratorformat_order_key)r
sort_orderprefixr r r ordert s z Sweep.orderc C s |du r| j }ntj|}|du rtd ntd| dd| jigi}ztj | j
| j | j ||ddd W S t
yC Y dS w ) zRReturn the best run sorted by the metric defined in config or the order passed in.NzVNo order specified and couldn't find metric in sweep config, returning most recent runzSorting runs by {}$andr& )r1 filtersper_pager )r1 r r- r. r r termlogr% r Runsr
r r
IndexError)r r1 r4 r r r best_run} s. zSweep.best_runreturnc C s | j dS )zJReturn the number of expected runs in the sweep or None for infinite runs.runCountExpected)r" r# r r r r expected_run_count s zSweep.expected_run_countc C s4 t jt| jt jt| jt jt| jgS r )urllibparse
quote_plusstrr r r r r r r path s z
Sweep.pathc C s$ | j }|dd | jjd| S )N sweeps/)rA insertr
app_urljoin)r rA r r r url s z Sweep.urlc C s | j dp| jS )Nr, )r r# r r r r r r, s z
Sweep.namec K s |du r| j }|||d}|| d} z |j||d} W n ty0 | j}|j||d} Y nw | r?| dr?| d dsAdS | d d }
| |||||
d} tj||||ddd| jigid | _ | S )
z*Execute a query against the cloud backend.N)r r r, )variable_valuesr r& )r
r2 )r1 r5 r4 )
QUERYupdateexecute Exception LEGACY_QUERYr# r r7 r r ) clsr
r r sidr1 querykwargs variablesresponsesweep_responser& r r r r# s@
z Sweep.get c C sF | j d }d| d}d}|r|d7 }td}|d|d|d S )
z9Generate HTML containing an iframe displaying this sweep.z
?jupyter=truez border:none;width:100%;height:zpx; z
display:none;r& z )rH r
toggle_button)r heighthiddenrH styler0 r r r to_html s
z
Sweep.to_htmlc C s | S r )r] r r r r _repr_html_ s zSweep._repr_html_c C s d d| j| jddS )NzrD statez
Unknown State)r% rG rA r" r# r r r r __repr__ s zSweep.__repr__r )F)NNNNN)rW F) __name__
__module__ __qualname____doc__r rK rO r propertyr r r boolr r1 r9 r intr< rA rH r, classmethodr# r] r@ r^ r`
__classcell__r r r r r sJ
2
r )rd r= typingr wandb_gqlr r r
wandb.apisr wandb.apis.attrsr
wandb.sdk.libr SWEEP_FRAGMENTr r r r r s