Ejemplos de register_custom_envs en Python

Ejemplo n.º 1

def register_all_envs():
    register_mujoco_envs()
    # register_pygame_envs()
    register_custom_envs()

Ejemplo n.º 2

from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

import gym

from dmbrl.config.pointmass_base import PointmassBaseConfigModule
from multiworld.core.flat_goal_env import FlatGoalEnv
from multiworld.envs.pygame import register_custom_envs
import numpy as np
import tensorflow as tf

register_custom_envs()


class PointmassReachFixedPointConfigModule(PointmassBaseConfigModule):
    TASK_HORIZON = 100
    NTRAIN_ITERS = 100
    NROLLOUTS_PER_ITER = 1
    PLAN_HOR = 10
    MODEL_IN, MODEL_OUT = 4, 2
    GP_NINDUCING_POINTS = 200

    GOAL_NP = np.array([1, 1])

    def __init__(self):
        super().__init__()
        env = gym.make('Point2DFixedGoalEnv-v0')
        env = FlatGoalEnv(env, append_goal_to_obs=False)
        self.ENV = env