Update progress bar info on total number of reals

equinor · Sep 13, 2024 · 838861c · 838861c
1 parent 3495366
commit 838861c
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/src/ert/run_models/base_run_model.py b/src/ert/run_models/base_run_model.py
@@ -391,10 +391,11 @@ def get_memory_consumption(self) -> int:
 
     def _current_progress(self) -> tuple[float, int]:
         current_iter = max(list(self._iter_snapshot.keys()))
-        done_realizations = 0
+        done_realizations = self.active_realizations.count(False)
         all_realizations = self._iter_snapshot[current_iter].reals
         current_progress = 0.0
-        realization_count = len(all_realizations)
+        # realization_count = len(all_realizations)
+        realization_count = len(self.active_realizations)
 
         if all_realizations:
             for real in all_realizations.values():
@@ -404,7 +405,10 @@ def _current_progress(self) -> tuple[float, int]:
                 ]:
                     done_realizations += 1
 
-            realization_progress = float(done_realizations) / len(all_realizations)
+            # realization_progress = float(done_realizations) / len(all_realizations)
+            realization_progress = float(done_realizations) / len(
+                self.active_realizations
+            )
             current_progress = (
                 (current_iter + realization_progress) / self._total_iterations
                 if self._total_iterations != 1