Cache ensemble_state

src/ert/enkf_main.py

@@ -194,6 +194,7 @@ def sample_prior(
             )
             ensemble.save_parameters(parameter, realization_nr, ds)
 
+    ensemble.refresh_ensemble_state()
     logger.debug(f"sample_prior() time_used {(time.perf_counter() - t):.4f}s")
 
 

src/ert/libres_facade.py

@@ -136,6 +136,7 @@ def load_from_forward_model(
             ensemble,
             [r for r, active in enumerate(realisations) if active],
         )
+        ensemble.refresh_ensemble_state()
         _logger.debug(
             f"load_from_forward_model() time_used {(time.perf_counter() - t):.4f}s"
         )
@@ -173,6 +174,7 @@ def load_from_run_path(
             else:
                 _logger.error(f"Realization: {iens}, load failure: {message}")
 
+        ensemble.refresh_ensemble_state()
         return loaded
 
     def get_observations(self) -> "EnkfObs":

src/ert/run_models/base_run_model.py

@@ -574,6 +574,8 @@ async def run_ensemble_evaluator_async(
             self._end_queue.get()
             return []
         await evaluator_task
+        ensemble.refresh_ensemble_state()
+
         return evaluator_task.result()
 
     # This function needs to be there for the sake of testing that expects sync ee run

src/ert/storage/local_ensemble.py

@@ -420,6 +420,11 @@ def get_failure(self, realization: int) -> Optional[_Failure]:
             )
         return None
 
+    def refresh_ensemble_state(self) -> None:
+        self.get_ensemble_state.cache_clear()
+        self.get_ensemble_state()
+
+    @lru_cache  # noqa: B019
     def get_ensemble_state(self) -> List[Set[RealizationStorageState]]:
         """
         Retrieve the state of each realization within ensemble.

src/ert/storage/local_storage.py

@@ -148,6 +148,9 @@ def refresh(self) -> None:
         self._ensembles = self._load_ensembles()
         self._experiments = self._load_experiments()
 
+        for ens in self._ensembles.values():
+            ens.refresh_ensemble_state()
+
     def get_experiment(self, uuid: UUID) -> LocalExperiment:
         """
         Retrieves an experiment by UUID.

tests/ert/unit_tests/dark_storage/test_common.py

@@ -98,5 +98,6 @@ def test_data_for_key_returns_empty_gen_data_config(tmp_path):
             ),
             0,
         )
+        ensemble.refresh_ensemble_state()
         data = data_for_key(ensemble, "response@0")
         assert not data.empty