()
| 190 | |
| 191 | |
| 192 | def parse_args(): |
| 193 | args = argparse.ArgumentParser() |
| 194 | args.add_argument("--custom_cfg", type=str, default="configs/offline_inference.yaml") |
| 195 | args.add_argument("--tree_jsonl", type=str, required=True, default="saved tree jsonl file.") |
| 196 | |
| 197 | args = args.parse_args() |
| 198 | return args |
| 199 | |
| 200 | |
| 201 | if __name__ == '__main__': |