| 12 | from matplotlib import pyplot as plt |
| 13 | |
| 14 | def parse(args): |
| 15 | parser = argparse.ArgumentParser( |
| 16 | description='wandb_usage', formatter_class=argparse.RawDescriptionHelpFormatter) |
| 17 | parser.add_argument("--team_name", type=str) |
| 18 | parser.add_argument("--project_name", type=str) |
| 19 | parser.add_argument("--experiment_name", type=str) |
| 20 | parser.add_argument("--scenario_name", type=str) |
| 21 | parser.add_argument("--seed",type=int,default=0) |
| 22 | all_args = parser.parse_known_args(args)[0] |
| 23 | return all_args |
| 24 | |
| 25 | def test_matplot(args): |
| 26 | |