INNER CODE UNIT · Python
get_args
KodCode-AI/kodcode · pipeline/step1.1_gen_questions.py:17
def get_args():
# Experiment Settings
parser = argparse.ArgumentParser(description="Seed Question Generation Manager.")
# Generation Parameters
parser.add_argument("--total_prompts", type=int, default=1000, help="Total number of prompts to generate. If specified, repeat will be ignored.")
parser.add_argument('--mode', default="leetcode", type=str, choices=["leetcode", "algorithm", "data_structure", "package", "apps", "codeforces", "code_contests", "taco", "docs", "evol", "prefill", "diversify"])
parser.add_argument('--language', default="python", type=str, choices=["cpp", "java", "python", "go", "javascript", "typescript", "csharp", "rust"])
# System Settings
parser.add_argument("--output_folder", type=str, default="../demo")
parser.add_argument("--job_name", type=str, default=None, help="Job Name.")
parser.add_argument("--timestamp", type=int, default=int(time.time()), help="Timestamp for the job. Also used as the random seed.")
parser.add_argument("--seed", type=int, default=None, help="Random seed.")
# Below are for Mode: Diversify
parser.add_argument("--input_data", type=str, default=None, help="Path to input jsonl file for diversify mode.")
# Below are for Mode: Prefill