Fix precommit

mlco2 · Oct 14, 2023 · 2679dbf · 2679dbf
1 parent f7eea7e
commit 2679dbf
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 9 deletions.
diff --git a/codecarbon/emissions_tracker.py b/codecarbon/emissions_tracker.py
@@ -701,7 +701,8 @@ def _measure_power_and_energy(self) -> None:
         self._do_measurements()
         self._last_measured_time = time.time()
         self._measure_occurrence += 1
-        if (len(self._output_handlers) > 0
+        if (
+            len(self._output_handlers) > 0
             and self._api_call_interval != -1
             and self._measure_occurrence >= self._api_call_interval
         ):

diff --git a/tests/test_custom_handler.py b/tests/test_custom_handler.py
@@ -2,13 +2,8 @@
 import unittest
 from typing import List
 
-from codecarbon.emissions_tracker import (
-    EmissionsTracker,
-    track_emissions,
-)
-from codecarbon.output import (
-    BaseOutput, EmissionsData,
-)
+from codecarbon.emissions_tracker import EmissionsTracker, track_emissions
+from codecarbon.output import BaseOutput, EmissionsData
 
 
 def heavy_computation(run_time_secs: int = 3):
@@ -65,7 +60,9 @@ def dummy_train_model():
         self.verify_custom_handler_state(handler_0)
         self.verify_custom_handler_state(handler_1)
 
-    def verify_custom_handler_state(self, handler: CustomOutput, expected_lines=1) -> None:
+    def verify_custom_handler_state(
+        self, handler: CustomOutput, expected_lines=1
+    ) -> None:
         assert len(handler.log) == expected_lines
         results = handler.log[0]
         self.assertEqual(results.project_name, self.project_name)