From e8c8660e680fcb753b726fb1bb87cacb6d805889 Mon Sep 17 00:00:00 2001 From: Mathieu Dupont <108517594+mathieudpnt@users.noreply.github.com> Date: Mon, 17 Nov 2025 17:15:00 +0100 Subject: [PATCH] ValueError in case of several tz in heatmap --- src/post_processing/utils/filtering_utils.py | 2 +- src/post_processing/utils/plot_utils.py | 3 +++ 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/src/post_processing/utils/filtering_utils.py b/src/post_processing/utils/filtering_utils.py index a965e41..db1484f 100644 --- a/src/post_processing/utils/filtering_utils.py +++ b/src/post_processing/utils/filtering_utils.py @@ -181,7 +181,7 @@ def get_timezone(df: DataFrame) -> tzoffset | list[tzoffset]: timezones = {ts.tz for ts in df["start_datetime"] if ts.tz is not None} if len(timezones) == 1: return next(iter(timezones)) - return sorted(timezones, key=lambda tz: tz.utcoffset(None)) + return list(timezones) def reshape_timebin( diff --git a/src/post_processing/utils/plot_utils.py b/src/post_processing/utils/plot_utils.py index 9641f73..9fdcefb 100644 --- a/src/post_processing/utils/plot_utils.py +++ b/src/post_processing/utils/plot_utils.py @@ -199,6 +199,9 @@ def _prepare_timeline_plot( if show_rise_set: tz = get_timezone(df) + if isinstance(tz, list): + msg = "Several timezones not supported." + raise ValueError(msg) add_sunrise_sunset(ax, lat, lon, tz)