Adapt start_date before pivoting 🤦

WorldCereal · Aug 21, 2024 · 7c0e160 · 7c0e160
1 parent 750930b
commit 7c0e160
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/presto/inference.py b/presto/inference.py
@@ -402,12 +402,13 @@ def process_parquet(df: pd.DataFrame) -> pd.DataFrame:
         (((df["timestamp"] - df["end_date"]).dt.days + 365) / 30).round().astype(int)
     )
 
+    # Now reassign start_date to the actual subset counted back from end_date
+    df["start_date"] = df["end_date"] - pd.Timedelta(days=364)
+
     df_pivot = df[(df["valid_date_ind"] >= 0) & (df["valid_date_ind"] < 12)].pivot(
         index=index_columns, columns="valid_date_ind", values=feature_columns
     )
 
-    # Now reassign start_date to the actual subset counted back from end_date
-    df["start_date"] = df["end_date"] - pd.Timedelta(days=364)
     # ----------------------------------------------------------------------------
 
     if df_pivot.empty: