Allow p=0 and p=1 for Bernoulli distribution

fidlej · Dec 3, 2012 · d4e443a · d4e443a
1 parent 949da45
commit d4e443a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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);
 }