add seeding to random choices in substrate configurations

PiperOrigin-RevId: 688547963
Change-Id: I732368f402801f8ce6b679a88545530219ff4c64

examples/modular/environment/modules/player_traits_and_styles.py

@@ -177,30 +177,38 @@
 )
 
 
-def get_trait(flowery: bool = False) -> str:
+def get_trait(flowery: bool = False,
+              rng: random.Random | None = None) -> str:
   """Get a random personality trait from a preset list of traits.
 
   Args:
     flowery: if True then use complex and flowery traits, if false then use
       single word traits.
+    rng: a random number generator.
 
   Returns:
     trait: a string
   """
+  if rng is None:
+    rng = random.Random()
   if flowery:
-    return random.choice(FLOWERY_TRAITS)
+    return rng.choice(FLOWERY_TRAITS)
   else:
-    return random.choice(TRAITS)
+    return rng.choice(TRAITS)
 
 
-def get_conversation_style(player_name: str) -> str:
+def get_conversation_style(player_name: str,
+                           rng: random.Random | None = None) -> str:
   """Get a random conversation style from a preset list of styles.
 
   Args:
     player_name: name of the player who will be said to have the sampled style
       of conversation.
+    rng: a random number generator.
 
   Returns:
     style: a string
   """
-  return random.choice(CONVERSATION_STYLES).format(player_name=player_name)
+  if rng is None:
+    rng = random.Random()
+  return rng.choice(CONVERSATION_STYLES).format(player_name=player_name)

examples/modular/environment/modules/pub_coordination_capetown.py

@@ -197,9 +197,12 @@
 
 def sample_parameters(seed: int | None = None):
   """Samples a set of parameters for the world configuration."""
-  pubs = random.sample(list(PUB_PREFERENCES.keys()), NUM_PUBS)
-  pub_preferences = {k: PUB_PREFERENCES[k] for k in pubs}
+
   seed = seed if seed is not None else random.getrandbits(63)
+  rng = random.Random(seed)
+
+  pubs = rng.sample(list(PUB_PREFERENCES.keys()), NUM_PUBS)
+  pub_preferences = {k: PUB_PREFERENCES[k] for k in pubs}
 
   config = pub_coordination.WorldConfig(
       year=YEAR,
@@ -212,8 +215,6 @@ def sample_parameters(seed: int | None = None):
       random_seed=seed,
   )
 
-  rng = random.Random(config.random_seed)
-
   all_names = list(MALE_NAMES) + list(FEMALE_NAMES)
 
   rng.shuffle(all_names)

examples/modular/environment/modules/pub_coordination_london.py

@@ -280,9 +280,11 @@
 
 def sample_parameters(seed: int | None = None):
   """Samples a set of parameters for the world configuration."""
-  pubs = random.sample(list(PUB_PREFERENCES.keys()), NUM_PUBS)
-  pub_preferences = {k: PUB_PREFERENCES[k] for k in pubs}
   seed = seed if seed is not None else random.getrandbits(63)
+  rng = random.Random(seed)
+
+  pubs = rng.sample(list(PUB_PREFERENCES.keys()), NUM_PUBS)
+  pub_preferences = {k: PUB_PREFERENCES[k] for k in pubs}
 
   config = pub_coordination.WorldConfig(
       year=YEAR,
@@ -294,7 +296,6 @@ def sample_parameters(seed: int | None = None):
       social_context=SOCIAL_CONTEXT,
       random_seed=seed,
   )
-  rng = random.Random(config.random_seed)
 
   all_names = list(MALE_NAMES) + list(FEMALE_NAMES)