Enabled amd-smi auto completion using argcomplete

Signed-off-by: Maisam Arif <maisarif@amd.com>
Change-Id: I0ea48b1b3a10e738d1fe91db85bc61ad51b85de2


[ROCm/amdsmi commit: d6abcfa92d]
Этот коммит содержится в:
Maisam Arif
2023-10-12 09:57:24 -05:00
коммит произвёл Dmitrii Galantsev
родитель 70558b418a
Коммит ef2c6cb157
+12
Просмотреть файл
@@ -1,4 +1,6 @@
#!/usr/bin/env python3
# PYTHON_ARGCOMPLETE_OK
#
# Copyright (C) 2023 Advanced Micro Devices. All rights reserved.
#
@@ -23,6 +25,11 @@
import logging
import sys
try:
import argcomplete
except ImportError:
logging.debug("argcomplete module not found. Autocomplete will not work.")
from amdsmi_commands import AMDSMICommands
from amdsmi_parser import AMDSMIParser
from amdsmi_logger import AMDSMILogger
@@ -63,6 +70,11 @@ if __name__ == "__main__":
amd_smi_commands.reset,
amd_smi_commands.rocm_smi)
try:
try:
argcomplete.autocomplete(amd_smi_parser)
except NameError:
logging.debug("argcomplete module not found. Autocomplete will not work.")
args = amd_smi_parser.parse_args(args=None if sys.argv[1:] else ['--help'])
# Handle command modifiers before subcommand execution