(all_args)
| 22 | return all_args |
| 23 | |
| 24 | def set_wandb(all_args): |
| 25 | run_dir = Path("../../../wandb_results") / all_args.project_name / all_args.experiment_name |
| 26 | if not run_dir.exists(): |
| 27 | os.makedirs(str(run_dir)) |
| 28 | |
| 29 | os.environ["WANDB_ENTITY"] = all_args.team_name |
| 30 | os.environ["WANDB_PROJECT"] = all_args.project_name |
| 31 | os.environ["WANDB_DIR"] = str(run_dir) |
| 32 | return run_dir |
| 33 | |
| 34 | def test_upload_artifact(run): |
| 35 | datasets = load() # separate code for loading the datasets |