增加windows 实现

eosphoros-ai · Oct 13, 2023 · 300b01a · 300b01a
1 parent 4a4d042
commit 300b01a
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/pilot/utils/system_utils.py b/pilot/utils/system_utils.py
@@ -206,9 +206,11 @@ def get_cpu_info():
             pass
     elif os_type == OSType.WINDOWS:
         try:
-            output = subprocess.check_output('wmic cpu get Name', shell=True).decode("utf-8")
+            output = subprocess.check_output("wmic cpu get Name", shell=True).decode(
+                "utf-8"
+            )
             lines = output.splitlines()
-            cpu_info = lines[2].split(':')[-1].strip()
+            cpu_info = lines[2].split(":")[-1].strip()
         except:
             pass