From badfc2a592905bcb669c8ef6e8261fe85735e3ad Mon Sep 17 00:00:00 2001 From: Ilia Dobrusin Date: Mon, 12 Oct 2020 00:17:45 +0200 Subject: [PATCH] Add kwargs to env creation to be compatible with api --- a2c_ppo_acktr/envs.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/a2c_ppo_acktr/envs.py b/a2c_ppo_acktr/envs.py index 1f1df6db8..f6b28b177 100755 --- 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)