[fix] Fix typos in stat.py to collect gpu memory and power correctly (#…

…3044)
aimhubio · Feb 6, 2024 · ac2ad12 · ac2ad12
1 parent cf46d7e
commit ac2ad12
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/aim/ext/resource/stat.py b/aim/ext/resource/stat.py
@@ -161,9 +161,7 @@ def get_stats(self):
                     memory = nvml.nvmlDeviceGetMemoryInfo(handle)
                     # Device memory usage
                     # 'memory_used': round10e5(memory.used / 1024 / 1024),
-                    gpu_info["gpu_memory_percent"] = (
-                        round10e5(memory.used * 100 / memory.total),
-                    )
+                    gpu_info["gpu_memory_percent"] = round10e5(memory.used * 100 / memory.total)
                 except nvml.NVMLError_NotSupported:
                     pass
                 try:
@@ -181,7 +179,7 @@ def get_stats(self):
                     power_cap_watts = power_cap / 1000
                     power_watts / power_cap_watts * 100
                     # Power usage in watts and percent
-                    gpu_info["gpu_power_watts"]: round10e5(power_watts)
+                    gpu_info["gpu_power_watts"] = round10e5(power_watts)
                     # gpu_info["power_percent"] = round10e5(power_usage)
                 except nvml.NVMLError_NotSupported:
                     pass