Add rng seed for scripts

Urban-Analytics-Technology-Platform · May 3, 2024 · ebec8b7 · ebec8b7
1 parent f9e3cbb
commit ebec8b7
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 0 deletions.
diff --git a/scripts/1_prep_synthpop.py b/scripts/1_prep_synthpop.py
@@ -1,7 +1,13 @@
+import numpy as np
 from uatk_spc.builder import Builder
 
+SEED = 0
+
 
 def main():
+    # Seed RNG
+    np.ranmdom.seed(SEED)
+
     # Pick a region with SPC output saved
     path = "../data/external/spc_output/raw/"
     region = "west-yorkshire"

diff --git a/scripts/2.1_sandbox-match_households.py b/scripts/2.1_sandbox-match_households.py
@@ -15,6 +15,9 @@
 
 pd.set_option("display.max_columns", None)
 
+# Seed RNG
+SEED = 0
+np.ranmdom.seed(SEED)
 
 # ## Step 1: Load in the datasets
 

diff --git a/scripts/2_match_households_and_individuals.py b/scripts/2_match_households_and_individuals.py
@@ -18,6 +18,10 @@
     truncate_values,
 )
 
+# Seed RNG
+SEED = 0
+np.ranmdom.seed(SEED)
+
 pd.set_option("display.max_columns", None)