Pandas 2 compability for ParameterResponseCorrelation

HansKallekleiv · Dec 19, 2024 · a3c8ef1 · a3c8ef1
1 parent 2af9699
commit a3c8ef1
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/webviz_subsurface/_utils/parameter_response.py b/webviz_subsurface/_utils/parameter_response.py
@@ -37,7 +37,9 @@ def filter_and_sum_responses(
     if aggregation == "sum":
         return df.groupby("REAL").sum().reset_index()[["REAL", response]]
     if aggregation == "mean":
-        return df.groupby("REAL").mean().reset_index()[["REAL", response]]
+        return (
+            df.groupby("REAL").mean(numeric_only=True).reset_index()[["REAL", response]]
+        )
     raise ValueError(f"Unknown aggregation '{aggregation}'.")