diff --git a/codecarbon/emissions_tracker.py b/codecarbon/emissions_tracker.py index 96cb16c4a..a4ffcbe9a 100644 --- 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 index 18d2c6a1b..44f11eb30 100644 --- 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)