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

Co-authored-by: Colt Allen <10178857+ColtAllen@users.noreply.github.com>
pymc-labs · Sep 10, 2024 · 690a79a · 690a79a
1 parent df0abcb
commit 690a79a
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: