fixup! Combine responses across realizations

equinor · Nov 26, 2024 · d3e8b2f · d3e8b2f
1 parent 8b31351
commit d3e8b2f
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/ert/storage/local_ensemble.py b/src/ert/storage/local_ensemble.py
@@ -733,7 +733,14 @@ def load_responses(self, key: str, realizations: Tuple[int]) -> polars.DataFrame
         combined_df_path = self.mount_point / f"{response_type}.parquet"
         if combined_df_path.exists():
             combined_df = polars.read_parquet(combined_df_path)
-            return combined_df.filter(polars.col("realization").is_in(realizations))
+            combined_df = combined_df.filter(
+                polars.col("realization").is_in(realizations)
+            )
+
+            if select_key:
+                combined_df = combined_df.filter(polars.col("response_key") == key)
+
+            return combined_df
 
         loaded = []
         for realization in realizations: