From f489939e1f0fe1207794511f6b2a3d7416aa42d0 Mon Sep 17 00:00:00 2001 From: Hans Kallekleiv <16436291+HansKallekleiv@users.noreply.github.com> Date: Thu, 19 Dec 2024 14:43:57 +0100 Subject: [PATCH] Pandas 2 compability for ParameterResponseCorrelation (#1310) --- webviz_subsurface/_utils/parameter_response.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/webviz_subsurface/_utils/parameter_response.py b/webviz_subsurface/_utils/parameter_response.py index c741eda62..b9054d32d 100644 --- 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}'.")