Merge pull request #764 from dirac-institute/rng_readability

Small change to getModuleRNG for readability
dirac-institute · Jan 5, 2024 · 0d681a8 · 0d681a8
2 parents b58fba3 + 76ef564
commit 0d681a8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/demo/test_bench.py b/demo/test_bench.py
@@ -35,7 +35,7 @@
     debug = False
     if debug:
         runLSSTSimulation(cmd_args_dict, configs)
-    else: # benchmark
+    else:  # benchmark
         cProfile.run("runLSSTSimulation(cmd_args_dict, configs)", "./tests/out/restats")
 
         p = pstats.Stats("./tests/out/restats")

diff --git a/src/sorcha/modules/PPModuleRNG.py b/src/sorcha/modules/PPModuleRNG.py
@@ -36,7 +36,8 @@ def getModuleRNG(self, module_name):
         if module_name in self._rngs:
             return self._rngs[module_name]
 
-        seed_offset = int(hashlib.md5(module_name.encode("utf-8")).hexdigest(), 16)
+        hashed_name = hashlib.md5(module_name.encode())
+        seed_offset = int(hashed_name.hexdigest(), base=16)
         module_seed = (self._base_seed + seed_offset) % (2**31)
         new_rng = np.random.default_rng(module_seed)
         self._rngs[module_name] = new_rng