LaunchAgent
Launch agent class which polls run given run queues and launches runs for wandb launch.
LaunchAgent(
api: Api,
config: Dict[str, Any]
)
| Arguments | |
|---|---|
api | Api object to use for making requests to the backend. |
config | Config dictionary for the agent. |
| Attributes | |
|---|---|
num_running_jobs | Return the number of jobs not including schedulers. |
num_running_schedulers | Return just the number of schedulers. |
thread_ids | Returns a list of keys running thread ids for the agent. |
Methods
check_sweep_state
check_sweep_state(
launch_spec, api
)
Check the state of a sweep before launching a run for the sweep.
fail_run_queue_item
fail_run_queue_item(
run_queue_item_id, message, phase, files=None
)
finish_thread_id
finish_thread_id(
thread_id, exception=None
)
Removes the job from our list for now.
get_job_and_queue
get_job_and_queue()
initialized
@classmethod
initialized() -> bool
Return whether the agent is initialized.
loop
loop()
Loop infinitely to poll for jobs and run them.
| Raises | |
|---|---|
KeyboardInterrupt | if the agent is requested to stop. |
name
@classmethod
name() -> str
Return the name of the agent.
pop_from_queue
pop_from_queue(
queue
)
Pops an item off the runqueue to run as a job.
| Arguments | |
|---|---|
queue | Queue to pop from. |
| Returns | |
|---|---|
| Item popped off the queue. |
| Raises | |
|---|---|
Exception | if there is an error popping from the queue. |
print_status
print_status() -> None
Prints the current status of the agent.
run_job
run_job(
job, queue, file_saver
)
Set up project and run the job.
| Arguments | |
|---|---|
job | Job to run. |
task_run_job
task_run_job(
launch_spec, job, default_config, api, job_tracker
)
update_status
update_status(
status
)
Update the status of the agent.
| Arguments | |
|---|---|
status | Status to update the agent to. |