def parse_args()

in data_preparation/generate_dataset.py [0:0]

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

def parse_args():
    parser = argparse.ArgumentParser(
        description="Prepare a dataset for InstructPix2Pix style training."
    )
    parser.add_argument(
        "--model_id", type=str, default="sayakpaul/whitebox-cartoonizer"
    )
    parser.add_argument("--dataset_id", type=str, default="imagenette")
    parser.add_argument("--max_num_samples", type=int, default=5000)
    parser.add_argument("--data_root", type=str, default="cartoonizer-dataset")
    args = parser.parse_args()
    return args