matthiasplappert

import numpy as np
import gym


env = gym.make('FetchReach-v0')

# Simply wrap the goal-based environment using FlattenDictWrapper
# and specify the keys that you would like to use.
env = gym.wrappers.FlattenDictWrapper(
    env, dict_keys=['observation', 'desired_goal'])