Use ensemble evaluator in everest in place of BatchSimulator

equinor · Nov 26, 2024 · 1def7b7 · 1def7b7
1 parent ca1806e
commit 1def7b7
Show file tree

Hide file tree

Showing 21 changed files with 618 additions and 1,828 deletions.
diff --git a/src/ert/__init__.py b/src/ert/__init__.py
@@ -15,14 +15,12 @@
 )
 from .data import MeasuredData
 from .libres_facade import LibresFacade
-from .simulator import BatchSimulator, BatchContext, JobStatus
+from .simulator import JobStatus
 from .workflow_runner import WorkflowRunner
 from .plugins import plugin
 from .scheduler import JobState
 
 __all__ = [
-    "BatchContext",
-    "BatchSimulator",
     "ErtScript",
     "ForwardModelStepDocumentation",
     "ForwardModelStepJSON",

diff --git a/src/ert/run_models/base_run_model.py b/src/ert/run_models/base_run_model.py
@@ -413,6 +413,12 @@ def _get_number_of_finished_realizations_from_reruns(self) -> int:
             False
         ) - self._initial_realizations_mask.count(False)
 
+    def get_current_snapshot(self) -> Optional[EnsembleSnapshot]:
+        if self._iter_snapshot.keys():
+            current_iter = max(list(self._iter_snapshot.keys()))
+            return self._iter_snapshot[current_iter]
+        return None
+
     def get_memory_consumption(self) -> int:
         max_memory_consumption: int = 0
         if self._iter_snapshot.keys():