Merge branch 'main' into patch-2

Author: asmith26

vose_sampler/vose_sampler.py

@@ -20,9 +20,10 @@ class VoseAlias(object):
     http://www.keithschwarz.com/darts-dice-coins/).
     """
 
-    def __init__(self, dist):
-        """ (VoseAlias, dict) -> NoneType """
+    def __init__(self, dist, rng=None):
+        """ (VoseAlias, dict[, RNG]) -> NoneType """
         self.dist = dist
+        self.rng = rng or random.Random()
         self.alias_initialisation()
         self.table_prob_list = list(self.table_prob)
 
@@ -68,10 +69,10 @@ def alias_initialisation(self):
     def alias_generation(self):
         """ Return a random outcome from the distribution. """
         # Determine which column of table_prob to inspect
-        col = random.choice(self.table_prob_list)
+        col = self.rng.choice(self.table_prob_list)
 
         # Determine which outcome to pick in that column
-        if self.table_prob[col] >= random.random():
+        if self.table_prob[col] >= self.rng.random():
             return col
         else:
             return self.table_alias[col]