Passer au contenu principal

Voir le code source

Classe de l’agent Launch qui interroge les files d’attente de run spécifiées et lance des Runs pour wandb Launch.
LaunchAgent(
    api: Api,
    config: Dict[str, Any]
)
ArgumentsDescription
apiObjet API à utiliser pour adresser des requêtes au backend.
configDictionnaire de configuration de l’agent.
AttributesDescription
num_running_jobsRenvoie le nombre de jobs, sans inclure les planificateurs.
num_running_schedulersRenvoie uniquement le nombre de planificateurs.
thread_idsRenvoie une liste des clés correspondant aux ID des threads en cours d’exécution pour l’agent.

Méthodes

check_sweep_state

Voir la source
check_sweep_state(
    launch_spec, api
)
Vérifiez l’état d’un balayage avant de lancer un run lié à ce balayage.

fail_run_queue_item

Voir la source
fail_run_queue_item(
    run_queue_item_id, message, phase, files=None
)

finish_thread_id

Voir la source
finish_thread_id(
    thread_id, exception=None
)
Retire le job de notre liste pour l’instant.

get_job_and_queue

Voir la source
get_job_and_queue()

initialized

Voir la source
@classmethod
initialized() -> bool
Indique si l’agent est initialisé.

loop

Voir la source
loop()
Effectue une boucle infinie pour interroger les jobs et les exécuter.
Exceptions levéesDescription
KeyboardInterruptsi l’agent reçoit une demande d’arrêt.

name

Voir la source
@classmethod
name() -> str
Renvoie le nom de l’agent.

pop_from_queue

Voir la source
pop_from_queue(
    queue
)
Extrait un élément de la runqueue pour l’exécuter comme job.
ArgumentsDescription
queueFile d’attente depuis laquelle extraire l’élément.
RenvoieDescription
Élément extrait de la file d’attente.
Exceptions levéesDescription
Exceptionen cas d’erreur lors de l’extraction depuis la file d’attente.
Voir la source
print_status() -> None
Affiche le statut actuel de l’agent.

run_job

Voir la source
run_job(
    job, queue, file_saver
)
Configurez le projet et exécutez le job.
ArgumentsDescription
jobJob à exécuter.

task_run_job

Voir la source
task_run_job(
    launch_spec, job, default_config, api, job_tracker
)

update_status

Voir la source
update_status(
    status
)
Mettez à jour le statut de l’agent.
ArgumentsDescription
statusStatut à attribuer à l’agent.