From bc8be8f3d87ad5559293a94b9333fd754b9711d8 Mon Sep 17 00:00:00 2001 From: Stephan de Hoop Date: Thu, 19 Dec 2024 13:09:27 +0100 Subject: [PATCH] Inline case_name variable (#9594) Remove casename --- src/everest/export.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/src/everest/export.py b/src/everest/export.py index 69ae317368c..ac49b082d92 100644 --- a/src/everest/export.py +++ b/src/everest/export.py @@ -153,12 +153,11 @@ def get_internalized_keys( internal_keys: set = set() with open_storage(storage_path, "r") as storage: for batch_id in batch_ids: - case_name = f"batch_{batch_id}" experiments = [*storage.experiments] assert len(experiments) == 1 experiment = experiments[0] - ensemble = experiment.get_ensemble_by_name(case_name) + ensemble = experiment.get_ensemble_by_name(f"batch_{batch_id}") if not internal_keys: internal_keys = set( ensemble.experiment.response_type_to_response_keys["summary"] @@ -316,14 +315,13 @@ def load_simulation_data( with open_storage(ens_path, "r") as storage: # pylint: disable=unnecessary-lambda-assignment def load_batch_by_id(): - case_name = f"batch_{batch}" experiments = [*storage.experiments] # Always assume 1 experiment per simulation/enspath, never multiple assert len(experiments) == 1 experiment = experiments[0] - ensemble = experiment.get_ensemble_by_name(case_name) + ensemble = experiment.get_ensemble_by_name(f"batch_{batch}") realizations = ensemble.get_realization_list_with_responses() try: df_pl = ensemble.load_responses("summary", tuple(realizations))