From fc23c78c77d8b952931d8aacec25f235712f64df Mon Sep 17 00:00:00 2001 From: Peter Zhokhov Date: Fri, 8 Nov 2019 15:39:57 -0800 Subject: [PATCH] fix imports --- baselines/common/cmd_util.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/baselines/common/cmd_util.py b/baselines/common/cmd_util.py index 86f6248..bd6ef9b 100644 --- a/baselines/common/cmd_util.py +++ b/baselines/common/cmd_util.py @@ -9,7 +9,7 @@ except ImportError: MPI = None import gym -from gym.wrappers import FlattenDictWrapper +from gym.wrappers import FlattenObservation, FilterObservation from baselines import logger from baselines.bench import Monitor from baselines.common import set_global_seeds @@ -81,7 +81,7 @@ def make_env(env_id, env_type, mpi_rank=0, subrank=0, seed=None, reward_scale=1. env = gym.make(env_id, **env_kwargs) if flatten_dict_observations and isinstance(env.observation_space, gym.spaces.Dict): - env = gym.wrappers.FlattenObservation(env) + env = FlattenObservation(env) env.seed(seed + subrank if seed is not None else None) env = Monitor(env, @@ -127,7 +127,7 @@ def make_robotics_env(env_id, seed, rank=0): """ set_global_seeds(seed) env = gym.make(env_id) - env = FlattenDictWrapper(env, ['observation', 'desired_goal']) + env = FlattenObservation(FilterObservation(env, ['observation', 'desired_goal'])) env = Monitor( env, logger.get_dir() and os.path.join(logger.get_dir(), str(rank)), info_keywords=('is_success',))