Fix related to column renaming after aggregating test frequency (#698)

Co-authored-by: Colt Allen <10178857+ColtAllen@users.noreply.github.com>
pymc-labs · May 28, 2024 · 3feec18 · 3feec18
1 parent 0e6ce83
commit 3feec18
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pymc_marketing/clv/utils.py b/pymc_marketing/clv/utils.py
@@ -563,7 +563,7 @@ def rfm_train_test_split(
     ).reset_index()
 
     test_rfm_data = test_rfm_data.rename(
-        columns={"id": "customer_id", "date": "test_frequency"}
+        columns={customer_id_col: "customer_id", datetime_col: "test_frequency"}
     )
 
     if monetary_value_col: