def parse_args()

in generate/run_ioi_slurm.py [0:0]

• 23 lines of code
• 1 McCabe index (conditional complexity)

def parse_args():
    parser = argparse.ArgumentParser(description="Run IOI evaluation on a model using Slurm")
    parser.add_argument("--model", type=str, required=True,
                        help="Model to evaluate (predefined model name)")
    parser.add_argument("--eval_args", type=str, required=True,
                        help="Arguments to pass to the evaluation script")
    parser.add_argument("--time", type=str, default="7-00:00:00",
                        help="Job time limit (default: 7 days)")
    parser.add_argument("--partition", type=str, default="hopper-prod",
                        help="Slurm partition")
    parser.add_argument("--qos", type=str, default="normal",
                        help="Slurm QOS")
    parser.add_argument("--startup_delay", type=int, default=3600,
                        help="Delay in seconds before starting the server")
    parser.add_argument("--dry_run", action="store_true",
                        help="Generate script but don't submit job")

    parser.add_argument("--revision", type=str, default=None, help="Revision to use for the model")
    parser.add_argument("--concurrency", type=int, default=100,
                        help="Number of concurrent requests to the server")
    
    parser.add_argument("--uv_env", type=str, default=None, help="Path to the uv env")
    parser.add_argument("--logs_dir", type=str, default=None)
    parser.add_argument("--slurm_dir", type=str, default=None)
    
    return parser.parse_args()