def get_args()

in prompts/stories/build_ultrachat_stories_prompts.py [0:0]

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

def get_args():
    parser = argparse.ArgumentParser()
    parser.add_argument("--repo_id", type=str, default="HuggingFaceTB/ultrachat_stories_prompts")
    parser.add_argument("--generation_style", type=str, default="textbook_academic")
    parser.add_argument("--run_all_styles", action="store_true")
    return parser.parse_args()