From 28b056732c182c31685a6a9b873e712021210b2e Mon Sep 17 00:00:00 2001 From: Peter Zhokhov Date: Fri, 16 Nov 2018 16:42:11 -0800 Subject: [PATCH] add flatten dict wrapper when observation space is dict (hack, don't merge) --- baselines/common/cmd_util.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/baselines/common/cmd_util.py b/baselines/common/cmd_util.py index 90b9868..8e110b5 100644 --- a/baselines/common/cmd_util.py +++ b/baselines/common/cmd_util.py @@ -53,6 +53,8 @@ def make_env(env_id, env_type, subrank=0, seed=None, reward_scale=1.0, gamestate env = retro_wrappers.make_retro(game=env_id, max_episode_steps=10000, use_restricted_actions=retro.Actions.DISCRETE, state=gamestate) else: env = gym.make(env_id) + if isinstance(env.observation_space, gym.spaces.Dict): + env = gym.wrappers.FlattenDictWrapper(env, dict_keys=['observation', 'desired_goal']) env.seed(seed + subrank if seed is not None else None) env = Monitor(env,