forgot to lint 🤪

ml-energy · Apr 30, 2024 · 9ab0c3c · 9ab0c3c
1 parent 4b50fba
commit 9ab0c3c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/zeus/device/gpu.py b/zeus/device/gpu.py
@@ -595,7 +595,9 @@ def supportsGetTotalEnergyConsumption(self) -> bool:
                 _ = amdsmi.amdsmi_get_energy_count(self.handle)
                 self._supportsGetTotalEnergyConsumption = True
             except amdsmi.AmdSmiLibraryException as e:
-                if e.get_error_code() == 2: # amdsmi.amdsmi_wrapper.AMDSMI_STATUS_NOT_SUPPORTED
+                if (
+                    e.get_error_code() == 2
+                ):  # amdsmi.amdsmi_wrapper.AMDSMI_STATUS_NOT_SUPPORTED
                     self._supportsGetTotalEnergyConsumption = False
                 else:
                     raise e