def parse_args()

in test.py [0:0]

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

def parse_args():
    parser = argparse.ArgumentParser(
        description="Evalaute a handwriting recognition model."
    )
    parser.add_argument(
        "--config", type=str, help="The json configuration file used for training."
    )
    parser.add_argument("--disable_cuda", action="store_true", help="Disable CUDA")
    parser.add_argument(
        "--checkpoint_path",
        default="/tmp/",
        type=str,
        help="Checkpoint path for loading the model",
    )
    parser.add_argument(
        "--load_last",
        default=False,
        action="store_true",
        help="Load the last saved model instead of the best model.",
    )
    parser.add_argument(
        "--split",
        default="validation",
        type=str,
        choices=["train", "validation", "test"],
        help="Data split to test on (default: 'validation')",
    )
    args = parser.parse_args()
    return args