Set env seeds whenever environment is changed

rlworkgroup · Nov 26, 2020 · 4ce934c · 4ce934c
1 parent 5417cd1
commit 4ce934c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/garage/sampler/_functions.py b/src/garage/sampler/_functions.py
@@ -1,5 +1,6 @@
 """Functions used by multiple Samplers or Workers."""
 from garage import Environment
+from garage.experiment import deterministic
 from garage.sampler.env_update import EnvUpdate
 
 
@@ -33,6 +34,7 @@ def _apply_env_update(old_env, env_update):
         elif isinstance(env_update, Environment):
             if old_env is not None:
                 old_env.close()
+            env_update.seed(deterministic.get_seed())
             return env_update, True
         else:
             raise TypeError('Unknown environment update type.')