Only send analysis time updates every second

equinor · Oct 24, 2024 · 50f9efe · 50f9efe
1 parent e47bc09
commit 50f9efe
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/src/ert/analysis/_es_update.py b/src/ert/analysis/_es_update.py
@@ -66,13 +66,16 @@ def noop_progress_callback(_: AnalysisEvent) -> None:
 
 
 class TimedIterator(Generic[T]):
+    SEND_FREQUENCY = 1.0  # seconds
+
     def __init__(
         self, iterable: Sequence[T], callback: Callable[[AnalysisEvent], None]
     ) -> None:
         self._start_time = time.perf_counter()
         self._iterable = iterable
         self._callback = callback
         self._index = 0
+        self._last_send_time = 0.0
 
     def __iter__(self) -> Self:
         return self
@@ -88,11 +91,14 @@ def __next__(self) -> T:
             estimated_remaining_time = (elapsed_time / (self._index)) * (
                 len(self._iterable) - self._index
             )
-            self._callback(
-                AnalysisTimeEvent(
-                    remaining_time=estimated_remaining_time, elapsed_time=elapsed_time
+            if elapsed_time - self._last_send_time > self.SEND_FREQUENCY:
+                self._callback(
+                    AnalysisTimeEvent(
+                        remaining_time=estimated_remaining_time,
+                        elapsed_time=elapsed_time,
+                    )
                 )
-            )
+                self._last_send_time = elapsed_time
 
         self._index += 1
         return result