info on gym + env.seed

PufferAI · Oct 29, 2023 · db0f7ad · db0f7ad
1 parent b1737c8
commit db0f7ad
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/pufferlib/environments/pokemon_red_updated.py b/pufferlib/environments/pokemon_red_updated.py
@@ -131,7 +131,11 @@ def __init__(
 
         self.reset()
 
-    def reset(self):
+    def reset(self, seed=None):
+        # Gym v26 favours `Env.reset(seed=seed)` this allows seeding to only be changed on environment reset.
+        # https://gymnasium.farama.org/content/migration-guide/#seed-and-random-number-generator
+        self.seed = seed
+
         # restart game, skipping credits
         self.initial_state.seek(0)
         self.pyboy.load_state(self.initial_state)