never mind, seem to need it

stisim/sim.py

@@ -7,10 +7,15 @@
 import sciris as sc
 import stisim as ss
 import itertools
-
+import numba as nb
 
 __all__ = ['Sim', 'AlreadyRunError', 'diff_sims']
 
+@nb.njit
+def set_numba_seed(value):
+    # Needed to ensure reproducibility when using random calls in numba, e.g. RandomNetwork
+    # Note, these random numbers are not currently common-random-number safe
+    np.random.seed(value)
 
 class Sim(sc.prettyobj):
 
@@ -79,6 +84,7 @@ def initialize(self, popdict=None, reset=False, **kwargs):
         self.validate_dt()
         self.init_time_vecs()  # Initialize time vecs
         ss.set_seed(self.pars['rand_seed'])  # Reset the random seed before the population is created
+        set_numba_seed(self.pars['rand_seed'])
 
         # Initialize the core sim components
         self.rng_container.initialize(self.pars['rand_seed'] + 2) # +2 ensures that seeds from the above population initialization and the +1-offset below are not reused within the rng_container

tests/benchmark.json

@@ -1,13 +1,13 @@
 {
   "time": {
     "people": 0.018,
-    "initialize": 0.043,
-    "run": 0.359
+    "initialize": 0.042,
+    "run": 0.356
   },
   "parameters": {
     "n_agents": 10000,
     "n_years": 20,
     "dt": 0.2
   },
-  "cpu_performance": 0.9410416196975259
+  "cpu_performance": 0.9653682555070681
 }