def parse_args()

in ember/cli.py [0:0]
77 lines of code
3 McCabe index (conditional complexity)

def parse_args(parser: argparse.ArgumentParser) -> argparse.ArgumentParser:
    subparsers = parser.add_subparsers(dest="command", required=True)

    generate_parser = subparsers.add_parser(
        "create", help="Create CoreML model for embeddings from a SentenceTransformer model."
    )

    generate_parser.add_argument(
        "--output-dir",
        type=str,
        default=".",
        help="Provide location to save exported models.",
    )
    generate_parser.add_argument(
        "--batch-size",
        type=int,
        default=64,
        help="Batch size for inference. Default is 64.",
    )
    generate_parser.add_argument(
        "--min-deployment-target",
        type=lambda x: getattr(AvailableTarget, x),
        choices=[target.name for target in AvailableTarget],
        default=AvailableTarget.iOS17,
        help="Minimum deployment target for CoreML model.",
    )
    generate_parser.add_argument(
        "--compute-units",
        type=lambda x: getattr(ComputeUnit, x),
        default=ComputeUnit.ALL,
        help="Which compute units to target for CoreML model.",
    )
    generate_parser.add_argument(
        "--precision",
        type=lambda x: getattr(ComputePrecision, x),
        choices=[p for p in ComputePrecision],
        default=ComputePrecision.FLOAT16,
        help="Precision used for computation in CoreML model, FLOAT16 can cause precision issues vs PyTorch model.",
    )
    generate_parser.add_argument(
        "--tolerance",
        type=float,
        default=5e-3,
        help="Tolerance for comparing PyTorch and CoreML model outputs.",
    )
    generate_parser.add_argument(
        "--psnr-threshold",
        type=float,
        default=35,
        help="PSNR threshold for comparing PyTorch and CoreML model outputs.",
    )
    generate_parser.add_argument(
        "--push-to-hub",
        action="store_true",
        help="Push the exported model to Hugging Face Hub",
        default=False,
    )
    generate_parser.add_argument(
        "--hf-repo-id",
        type=str,
        help="Hugging Face Hub repository ID (e.g., 'username/repo-name')",
        required=False,
    )
    generate_parser.add_argument(
        "--validate",
        action="store_true",
        help="Validate model outputs between PyTorch and CoreML",
        default=True,
    )
    generate_parser.add_argument(
        "--force-ane",
        action="store_true",
        help="Fail if model is not entirely ANE resident",
        default=False,
    )

    serve_parser = subparsers.add_parser(
        "serve", help="Serve the CoreML model locally."
    )

    return parser