From 0da6613b994a401bddf3e3288f08fe4e30c16c46 Mon Sep 17 00:00:00 2001 From: Maisam Arif Date: Fri, 4 Apr 2025 11:46:10 -0500 Subject: [PATCH] Updated imports on amdsmi_quick_start Signed-off-by: Maisam Arif Change-Id: Ideb0f2addf61fb6bdb728e549a8b0f133682d7d6 --- tools/amdsmi_quick_start.py | 22 ++++++++++++++++------ 1 file changed, 16 insertions(+), 6 deletions(-) diff --git a/tools/amdsmi_quick_start.py b/tools/amdsmi_quick_start.py index e12c67aee1..4c0c35a4b6 100644 --- a/tools/amdsmi_quick_start.py +++ b/tools/amdsmi_quick_start.py @@ -26,18 +26,28 @@ import logging import signal import sys -from amdsmi import * +try: + from amdsmi import * +except ImportError as e: + print(f"Failed to import 'amdsmi': {e}") + print("Ensure that the 'amdsmi' library is installed and accessible.") + raise from pathlib import Path sys.path.append(str(Path('/opt/rocm/libexec/'))) sys.path.append(str(Path('/opt/rocm/libexec/amdsmi_cli/'))) -from amdsmi_commands import AMDSMICommands -from amdsmi_helpers import AMDSMIHelpers -from amdsmi_logger import AMDSMILogger -from amdsmi_parser import AMDSMIParser -import amdsmi_cli_exceptions +try: + from amdsmi_commands import AMDSMICommands + from amdsmi_helpers import AMDSMIHelpers + from amdsmi_logger import AMDSMILogger + from amdsmi_parser import AMDSMIParser + import amdsmi_cli_exceptions +except ImportError as e: + print(f"Failed to import amdsmi cli libs: {e}") + print("Ensure that you have installed amdsmi's package.") + helpers = AMDSMIHelpers()