Update data

pipeline/transform/price_estimation/backtest.py

@@ -227,7 +227,18 @@ def main():
     fa = test["Total floor area (sqm)"].cast(pl.Float64).fill_null(0.0).to_numpy()
 
     print("\nComputing kNN estimates...")
-    knn_psm = knn_median_psm(trees, lat, lon, tg)
+    last_sale_dates = (
+        test["input_date"].dt.epoch("d").fill_null(-1).to_numpy().astype(np.int64)
+    )
+    knn_psm = knn_median_psm(
+        trees,
+        lat,
+        lon,
+        tg,
+        postcodes=test["Postcode"].fill_null("").to_numpy(),
+        last_prices=test["input_price"].cast(pl.Float64).to_numpy(),
+        last_sale_dates=last_sale_dates,
+    )
 
     # Temporal adjustment: pool PSM is at ref, adjust to actual
     log_idx_actual = test["log_index_actual"].to_numpy().astype(np.float64)