Skip to content

Commit

Permalink
Helper func to save runner details
Browse files Browse the repository at this point in the history
  • Loading branch information
romain-ragonnet committed Jul 27, 2023
1 parent a337301 commit d6b2d32
Showing 1 changed file with 18 additions and 0 deletions.
18 changes: 18 additions & 0 deletions autumn/projects/sm_covid2/common_school/runner_tools.py
Original file line number Diff line number Diff line change
Expand Up @@ -401,6 +401,24 @@ def opti_func(sample_dict):
Helper functions for remote runs
"""

def dump_runner_details(runner, out_folder_path):
"""
Dumps the run_path and IP associater with a SpringboardTaskRunner object
Args:
runner: SpringboardTaskRunner object
out_folder_path: pathlib Path associated with directory where data should be stored
"""

details = {
"run_path": runner.run_path,
"ip": runner.instance['ip']
}

out_path = out_folder_path / f"{runner.run_path.split('/')[-1]}.yml"
with out_path.open("w") as f:
yaml.dump(details, f)


def print_continuous_status(runner, update_freq=30):
status = runner.s3.get_status()
print(status)
Expand Down

0 comments on commit d6b2d32

Please sign in to comment.