diff --git a/lib/TH/THRandom.c b/lib/TH/THRandom.c index d34ed709..8470108b 100644 --- a/lib/TH/THRandom.c +++ b/lib/TH/THRandom.c @@ -233,6 +233,6 @@ int THRandom_geometric(double p) int THRandom_bernoulli(double p) { - THArgCheck(p > 0 && p < 1, 1, "must be > 0 and < 1"); + THArgCheck(p >= 0 && p <= 1, 1, "must be >= 0 and <= 1"); return(__uniform__() <= p); }