Merge pull request probmods#403 from null-a/add-bernoulli

Add bernoulli helper
droy · Apr 28, 2016 · 0c3863f · 0c3863f
2 parents 9044d2c + 77bdb0b
commit 0c3863f
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/src/header.wppl b/src/header.wppl
@@ -23,6 +23,11 @@ defineErpConstructors(
 
 // Helpers to sample from ERPs.
 
+var bernoulli = function(arg) {
+  var params = erp.isParams(arg) ? arg : {p: arg};
+  return sample(bernoulliERP(params));
+};
+
 var randomInteger = function(arg) {
   var params = erp.isParams(arg) ? arg : {n: arg};
   return sample(randomIntegerERP(params));
@@ -90,8 +95,8 @@ var gamma = function(arg1, arg2) {
 
 // Other ERP helpers
 
-var flip = function(arg) {
-  var params = erp.isParams(arg) ? arg : {p: (arg !== undefined) ? arg : .5};
+var flip = function(p) {
+  var params = {p: (p !== undefined) ? p : .5};
   return sample(bernoulliERP(params));
 };
 

diff --git a/tests/test-data/stochastic/models/flips.wppl b/tests/test-data/stochastic/models/flips.wppl
@@ -1,3 +1,3 @@
 var model = function() {
-  return flip() & flip(.2);
+  return flip() & bernoulli(.2);
 };