From f04d57d0d56fe821a4e9256df41dfc411c5a0ad7 Mon Sep 17 00:00:00 2001 From: xadupre Date: Sat, 22 Mar 2025 14:02:39 +0100 Subject: [PATCH] fix parser options --- onnx_array_api/_command_lines_parser.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/onnx_array_api/_command_lines_parser.py b/onnx_array_api/_command_lines_parser.py index e9b69a2..d1eac62 100644 --- a/onnx_array_api/_command_lines_parser.py +++ b/onnx_array_api/_command_lines_parser.py @@ -51,7 +51,7 @@ def get_parser_translate() -> ArgumentParser: parser.add_argument( "-a", "--api", - choices=["onnx", "light"], + choices=["onnx", "light", "onnx-short", "builder"], default="onnx", help="API to choose, API from onnx package or light API.", )