From 8b3135121fe4ee36a990b594f9726c955c6491ac Mon Sep 17 00:00:00 2001 From: "Yngve S. Kristiansen" Date: Tue, 26 Nov 2024 13:36:59 +0100 Subject: [PATCH] Route has_data through get_ensemble_state() --- src/ert/storage/local_ensemble.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/src/ert/storage/local_ensemble.py b/src/ert/storage/local_ensemble.py index e0d98edf87c..6e3241daa3b 100644 --- a/src/ert/storage/local_ensemble.py +++ b/src/ert/storage/local_ensemble.py @@ -291,14 +291,12 @@ def has_data(self) -> List[int]: exists : List[int] Returns the realization numbers with responses """ + + ensemble_state = self.get_ensemble_state() return [ i for i in range(self.ensemble_size) - if all( - (self._realization_dir(i) / f"{response}.parquet").exists() - or (config.has_finalized_keys and config.keys == []) - for response, config in self.experiment.response_configuration.items() - ) + if RealizationStorageState.RESPONSES_LOADED in ensemble_state[i] ] def realizations_initialized(self, realizations: List[int]) -> bool: