set random.seed if configured

NVIDIA · Oct 18, 2024 · 9fdd706 · 9fdd706
1 parent dc0bea8
commit 9fdd706
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/garak/cli.py b/garak/cli.py
@@ -330,6 +330,15 @@ def main(arguments=None) -> None:
     if "buffs" in args:
         _config.plugins.buff_spec = args.buffs
 
+    # base config complete
+
+    if hasattr(_config.run, "seed") and isinstance(_config.run.seed, int):
+        import random
+
+        random.seed(
+            _config.run.seed
+        )  # setting seed persists across re-imports of random
+
     # startup
     import sys
     import json