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/enum.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) (limited to 'ricky/param/enum.py') diff --git a/ricky/param/enum.py b/ricky/param/enum.py index d4a839a..137c58e 100644 --- a/ricky/param/enum.py +++ b/ricky/param/enum.py @@ -30,8 +30,10 @@ class Enum(Param): value = property(value_get, value_set) - def randomize(self): - self.value_set(random.choice(self._options)) + def randomize(self, probabilities=None): + if probabilities and self._choose_from_probabilities(probabilities): + return + self.value = random.choice(self._options) def as_dict(self): my_dict = super(Enum, self).as_dict() -- cgit v1.2.3-70-g09d2