Add seed parameter to stable_random_distribution

Author: Donaim

micall/utils/stable_random_distribution.py

@@ -3,16 +3,17 @@
 import numpy as np
 
 
-def stable_random_distribution(maximum: int) -> Iterator[int]:
+def stable_random_distribution(maximum: int, seed: int = 42) -> Iterator[int]:
     n = maximum
+    rng = np.random.default_rng(seed)
 
     weights = np.zeros(n) + 1
     forward = np.arange(1, n + 1)
     backwards = np.arange(n, 0, -1)
 
     while True:
         probabilities = weights / weights.sum()
-        index = np.random.choice(n, p=probabilities)
+        index = rng.choice(n, p=probabilities)
         yield index
         weights[:index] += forward[:index]
         weights[index:] += backwards[index:]