data.py

[robertpreseau]

Hey Nguyen,

I know this code is pretty old. I'm following your Math Behind A/B Testing walkthrough and can't get the B group to generate anything higher than A.

I changed B_bern = scs.bernoulli(p_B) to scs.bernoulli(p_A + p_B) and I'm getting numbers much closer to what you show in the tutorial.