fix parser options

xadupre · xadupre · commit f04d57d0d56f · 2025-03-22T14:02:39.000+01:00
diff --git 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.",
     )

Original file line number	Diff line number	Diff line change
`@@ -51,7 +51,7 @@ def get_parser_translate() -> ArgumentParser:`
`51`	`51`	`parser.add_argument(`
`52`	`52`	`"-a",`
`53`	`53`	`"--api",`
`54`		`- choices=["onnx", "light"],`
	`54`	`+ choices=["onnx", "light", "onnx-short", "builder"],`
`55`	`55`	`default="onnx",`
`56`	`56`	`help="API to choose, API from onnx package or light API.",`
`57`	`57`	`)`