| 370 | |
| 371 | |
| 372 | def parse_args(): |
| 373 | parser = argparse.ArgumentParser() |
| 374 | parser.add_argument("--jar_path", type=str, default="") |
| 375 | parser.add_argument("--task_nums", default="0") # use comma to split |
| 376 | parser.add_argument("--env_step_limit", type=int, default=100) |
| 377 | parser.add_argument("--simplification_str", default="easy") |
| 378 | parser.add_argument("--max_episode_per_file", type=int, default=9999) |
| 379 | parser.add_argument("--set", default="test") |
| 380 | parser.add_argument("--output_path", default="") |
| 381 | parser.add_argument("--no_stop", action="store_true", default=True) |
| 382 | parser.add_argument("--prompt_file", default="prompts/prompt.json") |
| 383 | parser.add_argument("--model_name", default="gpt-4") |
| 384 | |
| 385 | args = parser.parse_args() |
| 386 | params = vars(args) |
| 387 | return params |
| 388 | |
| 389 | # |
| 390 | # Main |