diff --git a/baselines/common/cmd_util.py b/baselines/common/cmd_util.py index 44dafa1..352eda2 100644 --- a/baselines/common/cmd_util.py +++ b/baselines/common/cmd_util.py @@ -32,7 +32,8 @@ def make_vec_env(env_id, env_type, num_env, seed, wrapper_kwargs=None, start_ind subrank = rank, seed=seed, reward_scale=reward_scale, - gamestate=gamestate + gamestate=gamestate, + wrapper_kwargs=wrapper_kwargs ) set_global_seeds(seed)