From c7a0c2781af52a0835646d406b387f961c22127b Mon Sep 17 00:00:00 2001 From: Peter Zhokhov Date: Tue, 23 Oct 2018 13:42:44 -0700 Subject: [PATCH] autopep8 and import fix --- baselines/registry.py | 4 ++-- baselines/run.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/baselines/registry.py b/baselines/registry.py index 3a99de6..b36b624 100644 --- a/baselines/registry.py +++ b/baselines/registry.py @@ -1,8 +1,8 @@ -# Registry of algorithms that keeps track of algorithms supported environments and +# Registry of algorithms that keeps track of algorithms supported environments and # and fine-grained defaults for different kinds of environments (atari, retro, mujoco etc) # # Example usage: -# +# # from baselines import registry # # @registry.register('fancy_algorithm', supports_vecenv=False) diff --git a/baselines/run.py b/baselines/run.py index 70bbff8..8ef8f50 100644 --- a/baselines/run.py +++ b/baselines/run.py @@ -95,7 +95,7 @@ def build_env(args): if registry[alg]['supports_vecenv']: env = make_vec_env(env_id, env_type, nenv, seed, gamestate=args.gamestate, reward_scale=args.reward_scale, frame_stack_size=frame_stack_size) else: - env = env_thunk(env_id, env_type, seed=seed, wrapper_kwargs={'frame_stack': frame_stack_size > 1}) + env = make_env(env_id, env_type, seed=seed, wrapper_kwargs={'frame_stack': frame_stack_size > 1}) if env_type == 'mujoco' and registry[alg]['supports_vecenv']: env = VecNormalize(env)