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': {