def create_quadrupped_env():
import gym
import roboschool
import quadruppedEnv
return wrappers.FrameStack(
wrappers.MaxAndSkipEnv(gym.make('QuadruppedWalk-v1'), 4, False), 2,
True)
def create_dm_control_env(**kwargs):
frames = kwargs.pop('frames', 1)
name = 'dm2gym:' + kwargs.pop('name')
env = gym.make(name, environment_kwargs=kwargs)
env = DMControlReward(env)
env = DMControlObsWrapper(env)
if frames > 1:
env = wrappers.FrameStack(env, frames, False)
return env
def create_goal_gym_env(**kwargs):
frames = kwargs.pop('frames', 1)
name = kwargs.pop('name')
limit_steps = kwargs.pop('limit_steps', False)
env = gym.make(name, **kwargs)
env = FlattenObservation(
FilterObservation(env, ['observation', 'desired_goal']))
if frames > 1:
env = wrappers.FrameStack(env, frames, False)
if limit_steps:
env = wrappers.LimitStepsWrapper(env)
return env
def create_default_gym_env(**kwargs):
frames = kwargs.pop('frames', 1)
name = kwargs.pop('name')
is_procgen = kwargs.pop('procgen', False)
limit_steps = kwargs.pop('limit_steps', False)
env = gym.make(name, **kwargs)
if frames > 1:
if is_procgen:
env = wrappers.ProcgenStack(env, frames, True)
else:
env = wrappers.FrameStack(env, frames, False)
if limit_steps:
env = wrappers.LimitStepsWrapper(env)
return env
'SuperMarioBrosRandomStage1-v1': {
'env_creator':
lambda **kwargs: create_super_mario_env_stage1(
'SuperMarioBrosRandomStage1-v1'),
'vecenv_type':
'RAY'
},
'RoboschoolHalfCheetah-v1': {
'env_creator':
lambda **kwargs: create_roboschool_env('RoboschoolHalfCheetah-v1'),
'vecenv_type':
'RAY'
},
'RoboschoolHumanoid-v1': {
'env_creator':
lambda: wrappers.FrameStack(
create_roboschool_env('RoboschoolHumanoid-v1'), 1, True),
'vecenv_type':
'RAY'
},
'LunarLanderContinuous-v2': {
'env_creator': lambda **kwargs: gym.make('LunarLanderContinuous-v2'),
'vecenv_type': 'RAY'
},
'RoboschoolHumanoidFlagrun-v1': {
'env_creator':
lambda **kwargs: wrappers.FrameStack(
create_roboschool_env('RoboschoolHumanoidFlagrun-v1'), 1, True),
'vecenv_type':
'RAY'
},
'BipedalWalker-v3': {
