added support for probabilities

1 files changed, 4 insertions, 2 deletions

diff --git a/ricky/param/bool.py b/ricky/param/bool.py
index 2b25bbd..4f75dc6 100644
--- a/ricky/param/bool.py
+++ b/ricky/param/bool.py
@@ -32,8 +32,10 @@ class Bool(Param):
             raise ValueError("Bad Value for Bool %s" % value)
         return value
 
-    def randomize(self):
-        self.value_set(random.choice([True, False]))
+    def randomize(self, probabilities=None):
+        if probabilities and self._choose_from_probabilities(probabilities):
+            return
+        self.value = random.choice([True, False])
 
     def from_normalized(self, value):
         value_as_int = int(round(value, 0))