Select for summary key in local_ensemble

yngve-sk · Oct 23, 2024 · ce1203e · ce1203e
1 parent 81463d6
commit ce1203e
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/src/ert/dark_storage/common.py b/src/ert/dark_storage/common.py
@@ -113,15 +113,14 @@ def data_for_key(
 
         if response_type == "summary":
             summary_data = ensemble.load_responses(
-                "summary",
-                tuple(ensemble.get_realization_list_with_responses("summary")),
+                response_key,
+                tuple(ensemble.get_realization_list_with_responses(response_key)),
             )
             if summary_data.is_empty():
                 return pd.DataFrame()
 
             df = (
-                summary_data.filter(polars.col("response_key").eq(key))
-                .rename({"time": "Date", "realization": "Realization"})
+                summary_data.rename({"time": "Date", "realization": "Realization"})
                 .drop("response_key")
                 .to_pandas()
             )