From 381de17e0a417f0c9234fa9b369ca033731ec141 Mon Sep 17 00:00:00 2001 From: amd-pworfolk Date: Fri, 6 Dec 2024 12:00:27 -0800 Subject: [PATCH] Fixed lint errors Signed-off-by: amd-pworfolk --- src/turnkeyml/common/system_info.py | 28 ++++++++++++---------------- 1 file changed, 12 insertions(+), 16 deletions(-) diff --git a/src/turnkeyml/common/system_info.py b/src/turnkeyml/common/system_info.py index 8439467..7308efa 100644 --- a/src/turnkeyml/common/system_info.py +++ b/src/turnkeyml/common/system_info.py @@ -1,9 +1,8 @@ +from abc import ABC +import importlib.metadata import platform -import subprocess import re -import importlib.metadata -import wmi -from abc import ABC +import subprocess class SystemInfo(ABC): @@ -59,6 +58,8 @@ class WindowsSystemInfo(SystemInfo): def __init__(self): super().__init__() + import wmi + self.connection = wmi.WMI() def get_processor_name(self) -> str: @@ -70,7 +71,11 @@ def get_processor_name(self) -> str: """ processors = self.connection.Win32_Processor() if processors: - return f"{processors[0].Name.strip()} ({processors[0].NumberOfCores} cores, {processors[0].NumberOfLogicalProcessors} logical processors)" + return ( + f"{processors[0].Name.strip()} " + f"({processors[0].NumberOfCores} cores, " + f"{processors[0].NumberOfLogicalProcessors} logical processors)" + ) return "Processor information not found." def get_system_model(self) -> str: @@ -166,7 +171,7 @@ def get_npu_power_mode() -> str: except FileNotFoundError: # xrt-smi not present pass - except subprocess.CalledProcessError as e: + except subprocess.CalledProcessError: pass return "NPU power mode not found." @@ -181,7 +186,7 @@ def get_windows_power_setting() -> str: try: out = subprocess.check_output(["powercfg", "/getactivescheme"]).decode() return re.search(r"\((.*?)\)", out).group(1) - except subprocess.CalledProcessError as e: + except subprocess.CalledProcessError: pass return "Windows power setting not found" @@ -219,9 +224,6 @@ def get_dict(self) -> dict: class WSLSystemInfo(SystemInfo): """Class used to access system information in WSL""" - def __init__(self): - super().__init__() - @staticmethod def get_system_model() -> str: """ @@ -261,9 +263,6 @@ def get_dict(self) -> dict: class LinuxSystemInfo(SystemInfo): """Class used to access system information in Linux""" - def __init__(self): - super().__init__() - @staticmethod def get_processor_name() -> str: """ @@ -345,9 +344,6 @@ def get_dict(self) -> dict: class UnsupportedOSSystemInfo(SystemInfo): """Class used to access system information in unsupported operating systems""" - def __init__(self): - super().__init__() - def get_dict(self): """ Retrieves all the system information into a dictionary