Fix ModifiedBetaGeoNBDRV sim_data (#1472)

* Fix ModifiedBetaGeoNBDRV sim_data * pre-commit.ci run * pre-commit.ci run
pymc-labs · Feb 6, 2025 · 15793e2 · 15793e2
1 parent 4ef55f3
commit 15793e2
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pymc_marketing/clv/distributions.py b/pymc_marketing/clv/distributions.py
@@ -783,7 +783,9 @@ def sim_data(lam, p, T):
             t = 0  # recency
             n = 0  # frequency
 
-            churn = p  # MBG/NBD customer active with probability p at time 0
+            churn = (
+                rng.random() < p
+            )  # MBG/NBD customer active with probability p at time 0
             wait = rng.exponential(scale=1 / lam)
 
             while t + wait < T and not churn: