Skip to content

Commit 861a68e

Browse files
committed
feat: add deprecated=True and deprecate_reason for argparse
Signed-off-by: Aaron Pham <contact@aarnphm.xyz>
1 parent 9352cdb commit 861a68e

File tree

1 file changed

+16
-0
lines changed

1 file changed

+16
-0
lines changed

vllm/utils.py

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1332,6 +1332,22 @@ def __init__(self, *args, **kwargs):
13321332
kwargs['formatter_class'] = SortedHelpFormatter
13331333
super().__init__(*args, **kwargs)
13341334

1335+
def add_argument(self, *args: Any, **kwargs: Any):
1336+
# add a deprecated=True with optional deprecated_reason to signify
1337+
# reasons for deprecating this args
1338+
if kwargs.pop("deprecated", False):
1339+
deprecated_message = kwargs.pop("deprecated_reason", None)
1340+
if 'help' in kwargs:
1341+
kwargs['help'] = (
1342+
f"[DEPRECATED]{(' ' + deprecated_message) or ''}.\n{kwargs['help']}" # noqa: E501
1343+
)
1344+
else:
1345+
kwargs['help'] = (
1346+
f"[DEPRECATED]{(' ' + deprecated_message) or ''}" # noqa: E501
1347+
)
1348+
1349+
super().add_argument(*args, **kwargs)
1350+
13351351
def parse_args(self, args=None, namespace=None):
13361352
if args is None:
13371353
args = sys.argv[1:]

0 commit comments

Comments
 (0)