def parse_args()

in validation/compare_models.py [0:0]

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

def parse_args():
    parser = argparse.ArgumentParser()
    parser.add_argument(
        "--model_id",
        type=str,
        default="sayakpaul/whitebox-cartoonizer",
        choices=[
            "sayakpaul/whitebox-cartoonizer",
            "instruction-tuning-vision/instruction-tuned-cartoonizer",
            "timbrooks/instruct-pix2pix",
        ],
    )
    parser.add_argument("--dataset_id", type=str, default="imagenette")
    parser.add_argument("--max_num_samples", type=int, default=10)
    parser.add_argument(
        "--prompt", type=str, default="Generate a cartoonized version of the image"
    )
    parser.add_argument("--num_inference_steps", type=int, default=20)
    parser.add_argument("--image_guidance_scale", type=float, default=1.5)
    parser.add_argument("--guidance_scale", type=float, default=7.0)
    args = parser.parse_args()
    return args