More precise start_date determination

WorldCereal · Aug 21, 2024 · 3f5cae0 · 3f5cae0
1 parent 7c0e160
commit 3f5cae0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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