Fix simulation context while fix the correct imports RealizationState…

… and RealizationStorageState
xjules · Nov 23, 2023 · 4ea0014 · 4ea0014
1 parent fb95b09
commit 4ea0014
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/ert/simulator/batch_simulator_context.py b/src/ert/simulator/batch_simulator_context.py
@@ -7,7 +7,7 @@
 
 import numpy as np
 
-from ert.realization_state import RealizationState
+from ert.job_queue import RealizationState
 
 from .simulation_context import SimulationContext
 

diff --git a/src/ert/simulator/simulation_context.py b/src/ert/simulator/simulation_context.py
@@ -10,7 +10,7 @@
 
 from ert.config import HookRuntime
 from ert.enkf_main import create_run_path
-from ert.job_queue import JobQueue, JobStatus
+from ert.job_queue import JobQueue, RealizationState
 from ert.run_context import RunContext
 from ert.runpaths import Runpaths
 from ert.storage.realization_storage_state import RealizationStorageState
@@ -68,12 +68,12 @@ def _run_forward_model(
     ):
         queue_evaluators = [
             partial(
-                job_queue.stop_long_running_jobs,
+                job_queue.stop_long_running_realizations,
                 ert.ert_config.analysis_config.minimum_required_realizations,
             )
         ]
 
-    asyncio.run(job_queue.execute(evaluators=queue_evaluators))
+    asyncio.run(job_queue.execute(evaluators=queue_evaluators))  # type: ignore
 
     run_context.sim_fs.sync()