def parse_args()

in distilvit/upload.py [0:0]

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

def parse_args():
    parser = argparse.ArgumentParser(
        description="Upload a model to the HF hub",
        formatter_class=argparse.ArgumentDefaultsHelpFormatter,
    )

    parser.add_argument(
        "--model-id",
        type=str,
        help="Model ID",
    )

    parser.add_argument(
        "--save-path",
        type=str,
        help="Save dir",
    )

    parser.add_argument(
        "--tag",
        type=str,
        help="HF tag",
        default=None,
    )

    parser.add_argument(
        "--commit-message",
        default="New Training",
        type=str,
        help="Save dir",
    )

    return parser.parse_args()