(args)
| 86 | |
| 87 | |
| 88 | def test_artifact(args): |
| 89 | all_args = parse(args) |
| 90 | run_dir = set_wandb(all_args) |
| 91 | |
| 92 | # 🚀 start a run, with a type to label it and a project it can call home |
| 93 | with wandb.init(entity=all_args.team_name, |
| 94 | project=all_args.project_name, |
| 95 | config=all_args, |
| 96 | dir=str(run_dir), |
| 97 | name=all_args.experiment_name + "_" + str(all_args.seed), |
| 98 | job_type="load-data") as run: |
| 99 | test_upload_artifact(run) |
| 100 | |
| 101 | count_down(10) |
| 102 | |
| 103 | test_download_artifact(run) |
| 104 | |
| 105 | if __name__ == '__main__': |
| 106 | test_artifact(sys.argv[1:]) |
no test coverage detected