Skip to content

Commit

Permalink
gen_data_keys -> gen_data_display_keys
Browse files Browse the repository at this point in the history
  • Loading branch information
yngve-sk committed Oct 23, 2024
1 parent 94059e9 commit 93fecc9
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 5 deletions.
6 changes: 3 additions & 3 deletions src/ert/dark_storage/common.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down Expand Up @@ -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
)
Expand Down
4 changes: 2 additions & 2 deletions src/ert/dark_storage/endpoints/records.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down Expand Up @@ -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,
Expand Down

0 comments on commit 93fecc9

Please sign in to comment.