(args, filename, exclude=None)
| 94 | |
| 95 | |
| 96 | def save_argparse(args, filename, exclude=None): |
| 97 | os.makedirs(dirname(filename), exist_ok=True) |
| 98 | if filename.endswith("yaml") or filename.endswith("yml"): |
| 99 | if isinstance(exclude, str): |
| 100 | exclude = [exclude] |
| 101 | args = args.__dict__.copy() |
| 102 | for exl in exclude: |
| 103 | del args[exl] |
| 104 | yaml.dump(args, open(filename, "w")) |
| 105 | else: |
| 106 | raise ValueError("Configuration file should end with yaml or yml") |
| 107 | |
| 108 | |
| 109 | def number(text): |