From 583ba082a2ade49455030f38374e889874b885fd Mon Sep 17 00:00:00 2001 From: pzhokhov Date: Tue, 23 Oct 2018 11:22:27 -0700 Subject: [PATCH] Update cmd_util.py --- baselines/common/cmd_util.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/baselines/common/cmd_util.py b/baselines/common/cmd_util.py index 352eda2..7c38a77 100644 --- a/baselines/common/cmd_util.py +++ b/baselines/common/cmd_util.py @@ -43,7 +43,7 @@ def make_vec_env(env_id, env_type, num_env, seed, wrapper_kwargs=None, start_ind return DummyVecEnv([make_thunk(start_index)]) -def make_env(env_id, env_type, subrank=0, seed=None, reward_scale=1.0, gamestate=None, wrapper_kwargs=None): +def make_env(env_id, env_type, subrank=0, seed=None, reward_scale=1.0, gamestate=None, wrapper_kwargs={}): mpi_rank = MPI.COMM_WORLD.Get_rank() if MPI else 0 if env_type == 'atari': env = make_atari(env_id)