From 3f5cae08f1be3d037ee01d18b4a960c822be29f8 Mon Sep 17 00:00:00 2001 From: Kristof Van Tricht Date: Wed, 21 Aug 2024 17:48:47 +0200 Subject: [PATCH] More precise start_date determination --- presto/inference.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/presto/inference.py b/presto/inference.py index 87c3321..122ece9 100644 --- a/presto/inference.py +++ b/presto/inference.py @@ -403,7 +403,7 @@ def process_parquet(df: pd.DataFrame) -> pd.DataFrame: ) # Now reassign start_date to the actual subset counted back from end_date - df["start_date"] = df["end_date"] - pd.Timedelta(days=364) + df["start_date"] = df["end_date"] - pd.DateOffset(years=1) + pd.DateOffset(days=1) df_pivot = df[(df["valid_date_ind"] >= 0) & (df["valid_date_ind"] < 12)].pivot( index=index_columns, columns="valid_date_ind", values=feature_columns