Add kwargs to env creation to be compatible with api

lukashermann · Oct 11, 2020 · badfc2a · badfc2a
1 parent 9ab2fa5
commit badfc2a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/a2c_ppo_acktr/envs.py b/a2c_ppo_acktr/envs.py
@@ -31,9 +31,9 @@
 #     pass
 
 
-def make_env(env_id, seed, rank, log_dir, add_timestep, allow_early_resets):
+def make_env(env_id, seed, rank, log_dir, add_timestep, allow_early_resets, **env_kwargs):
     def _thunk():
-        env = gym.make(env_id)
+        env = gym.make(env_id, **env_kwargs)
 
         env.seed(seed + rank)