diff --git a/projects/amdsmi/amdsmi_cli/amdsmi_cli.py b/projects/amdsmi/amdsmi_cli/amdsmi_cli.py index 3a88ae5acc..3bafe179ca 100755 --- a/projects/amdsmi/amdsmi_cli/amdsmi_cli.py +++ b/projects/amdsmi/amdsmi_cli/amdsmi_cli.py @@ -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