protomotions.agents.callbacks.slurm_autoresume_srun module#

protomotions.agents.callbacks.slurm_autoresume_srun.wandb_run_exists()[source]#

class protomotions.agents.callbacks.slurm_autoresume_srun.AutoResumeCallbackSrun(autoresume_after=12600)[source]#

Bases: <Mock object at 0x7fbf79e7f550>[]

__init__(autoresume_after=12600)[source]#

before_play_steps(agent)[source]#

on_fit_start(agent)[source]#

on_fit_end(agent)[source]#