From e1a642e41e7e6d08303573fb20f265b4ac326372 Mon Sep 17 00:00:00 2001 From: pepperpepperpepper Date: Mon, 7 Dec 2015 14:40:04 -0800 Subject: added support for probabilities --- ricky/param/bool.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) (limited to 'ricky/param/bool.py') 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)) -- cgit v1.2.3-70-g09d2