From 339c8a1ae472fad405da3696b161c375a8d6aa74 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Anton=20Jyrki=C3=A4inen?= Date: Mon, 8 Dec 2025 13:40:07 +0200 Subject: [PATCH] debug get_season --- python/common/recluster.py | 3 ++- python/common/utils.py | 5 +++++ 2 files changed, 7 insertions(+), 1 deletion(-) diff --git a/python/common/recluster.py b/python/common/recluster.py index 09e7a07..be29eaa 100644 --- a/python/common/recluster.py +++ b/python/common/recluster.py @@ -476,9 +476,10 @@ def ui_related_var_modifications(df: pd.DataFrame, seasons_and_months: dict, DEP pd.DataFrame: clusters with ui related variables """ df["tst_median"] = pd.to_datetime(df["tst_median"], format="%Y-%m-%d %H:%M:%S", errors="coerce") + print(df["tst_median"]) df["year"] = df["tst_median"].dt.year df["season"] = df["tst_median"].dt.month.map(lambda x: get_season(x, seasons_and_months)) - + print(df["season"]) for k, v in DCLASS_NAMES.items(): df["dclass"] = df["dclass"].replace(k, v) diff --git a/python/common/utils.py b/python/common/utils.py index e5fdf6f..71c2daf 100644 --- a/python/common/utils.py +++ b/python/common/utils.py @@ -53,6 +53,11 @@ def get_target_oday(offset=1): return start_date def get_season(month, seasons_and_months): + print(month) + print(seasons_and_months) + if month is None: + return None + key = [key for key, val in seasons_and_months.items() if month in val][0] return key.lower()