Pass lint

ml-energy · Aug 20, 2023 · 63ca271 · 63ca271
1 parent 4cccc10
commit 63ca271
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/zeus/monitor/__main__.py b/zeus/monitor/__main__.py
@@ -49,7 +49,8 @@ def energy(gpu_indices: Optional[list[int]] = None) -> None:
 
 @app.command()
 def power(
-    gpu_indices: Optional[list[int]] = None, update_period: float = 1.0,
+    gpu_indices: Optional[list[int]] = None,
+    update_period: float = 1.0,
 ) -> None:
     """Monitor the power consumption of GPUs during the duration of the CLI program.
 
@@ -68,7 +69,7 @@ def power(
         end_time = time.time()
         print(f"Total time: {end_time - start_time} s")
         print(f"Total energy: {monitor.get_energy(start_time, end_time)} J")
-        raise typer.Exit()
+        raise typer.Exit() from None
 
 
 if __name__ == "__main__":

diff --git a/zeus/monitor/power.py b/zeus/monitor/power.py
@@ -126,10 +126,8 @@ def __init__(
         # Spawn the power polling process.
         self.process = None
         atexit.register(self._stop)
-        self.process = (
-            mp
-            .get_context("spawn")
-            .Process(target=self._run, args=(gpu_indices, power_csv, update_period))
+        self.process = mp.get_context("spawn").Process(
+            target=self._run, args=(gpu_indices, power_csv, update_period)
         )
 
     def _run(