diff --git a/src/ert/dark_storage/common.py b/src/ert/dark_storage/common.py index b6e0c7960fc..ee00d770aba 100644 --- a/src/ert/dark_storage/common.py +++ b/src/ert/dark_storage/common.py @@ -78,11 +78,11 @@ def ensemble_parameters(storage: Storage, ensemble_id: UUID) -> List[Dict[str, A def get_response_names(ensemble: Ensemble) -> List[str]: result = ensemble.get_summary_keyset() - result.extend(sorted(gen_data_keys(ensemble), key=lambda k: k.lower())) + result.extend(sorted(gen_data_display_keys(ensemble), key=lambda k: k.lower())) return result -def gen_data_keys(ensemble: Ensemble) -> Iterator[str]: +def gen_data_display_keys(ensemble: Ensemble) -> Iterator[str]: gen_data_config = ensemble.experiment.response_configuration.get("gen_data") if gen_data_config: @@ -263,7 +263,7 @@ def get_observation_keys_for_response( Get all observation keys for given response key """ - if displayed_response_key in gen_data_keys(ensemble): + if displayed_response_key in gen_data_display_keys(ensemble): response_key, report_step = displayed_key_to_response_key["gen_data"]( displayed_response_key ) diff --git a/src/ert/dark_storage/endpoints/records.py b/src/ert/dark_storage/endpoints/records.py index 70be40d4070..2aacd9e93b5 100644 --- a/src/ert/dark_storage/endpoints/records.py +++ b/src/ert/dark_storage/endpoints/records.py @@ -12,7 +12,7 @@ from ert.dark_storage.common import ( data_for_key, ensemble_parameters, - gen_data_keys, + gen_data_display_keys, get_observation_keys_for_response, get_observations_for_obs_keys, response_key_to_displayed_key, @@ -141,7 +141,7 @@ def get_ensemble_responses( has_observations=name in response_names_with_observations, ) - for name in gen_data_keys(ensemble): + for name in gen_data_display_keys(ensemble): response_map[str(name)] = js.RecordOut( id=UUID(int=0), name=name,