From 88371cf93e70abbc2515fc3107c78653ab476512 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Anton=20Jyrki=C3=A4inen?= Date: Wed, 3 Dec 2025 18:43:40 +0200 Subject: [PATCH] reverting changes that caused KeyError: 'tst_median' --- python/common/recluster.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/python/common/recluster.py b/python/common/recluster.py index ae997ea..d283fad 100644 --- a/python/common/recluster.py +++ b/python/common/recluster.py @@ -113,9 +113,6 @@ async def load_preprocess_files( compressed_data = r[0] decompressed_csv = decompressor.decompress(compressed_data) df = pd.read_csv(io.BytesIO(decompressed_csv), sep=";") - df["tst_median"] = pd.to_datetime(df["tst_median"], format="ISO8601").dt.tz_convert( - "UTC" - ) dfs.append(df) if not dfs: @@ -438,7 +435,6 @@ def calculate_cluster_features(df: pd.DataFrame, cluster_id_vars_on_2nd_level: l res = res.merge(clust_delay_feats, on=cluster_id_vars_on_2nd_level, how="outer") res["oday_min"] = df.oday.min() res["oday_max"] = df.oday.max() - res["tst_median"] = res["tst_median"].dt.tz_convert("Europe/Helsinki") return res