Skip to main content

View the source code

 Launch agent class which polls run given run queues and launches runs for wandb launch. 
LaunchAgent(
    api: Api,
    config: Dict[str, Any]
)
ArgumentsDescription
apiApi object to use for making requests to the backend.
configConfig dictionary for the agent.
AttributesDescription
num_running_jobsReturn the number of jobs not including schedulers.
num_running_schedulersReturn just the number of schedulers.
thread_idsReturns a list of keys running thread ids for the agent.

​
Methods

​
check_sweep_state

View source 
check_sweep_state(
    launch_spec, api
)
Check the state of a sweep before launching a run for the sweep.

​
fail_run_queue_item

View source 
fail_run_queue_item(
    run_queue_item_id, message, phase, files=None
)

​
finish_thread_id

View source 
finish_thread_id(
    thread_id, exception=None
)
Removes the job from our list for now.

​
get_job_and_queue

View source 
get_job_and_queue()

​
initialized

View source 
@classmethod
initialized() -> bool
Return whether the agent is initialized.

​
loop

View source 
loop()
Loop infinitely to poll for jobs and run them.
RaisesDescription
KeyboardInterruptif the agent is requested to stop.

​
name

View source 
@classmethod
name() -> str
Return the name of the agent.

​
pop_from_queue

View source 
pop_from_queue(
    queue
)
Pops an item off the runqueue to run as a job.
ArgumentsDescription
queueQueue to pop from.
ReturnsDescription
Item popped off the queue.
RaisesDescription
Exceptionif there is an error popping from the queue.
View source 
print_status() -> None
Prints the current status of the agent.

​
run_job

View source 
run_job(
    job, queue, file_saver
)
Set up project and run the job.
ArgumentsDescription
jobJob to run.

​
task_run_job

View source 
task_run_job(
    launch_spec, job, default_config, api, job_tracker
)

​
update_status

View source 
update_status(
    status
)
Update the status of the agent.
ArgumentsDescription
statusStatus to update the agent to.
⌘I