import gym import gym.wrappers import gym.envs import gym.spaces import traceback import logging try: from gym.wrappers.monitoring import logger as monitor_logger monitor_logger.setLevel(logging.WARNING) except Exception as e: traceback.print_exc() import os import os.path as osp from rllab.envs.base import Env, Step from rllab.core.serializable import Serializable from rllab.spaces.box import Box from rllab.spaces.discrete import Discrete from rllab.spaces.product import Product from rllab.misc import logger def convert_gym_space(space): if isinstance(space, gym.spaces.Box): return Box(low=space.low, high=space.high) elif isinstance(space, gym.spaces.Discrete): return Discrete(n=space.n) elif isinstance(space, gym.spaces.Tuple): return Product([convert_gym_space(x) for x in space.spaces])
import gym import gym.wrappers import gym.envs import gym.spaces import traceback import logging try: from gym.wrappers.monitoring import logger as monitor_logger monitor_logger.setLevel(logging.WARNING) except Exception as e: traceback.print_exc() import os import os.path as osp from rllab.envs.base import Env, Step from rllab.core.serializable import Serializable from rllab.spaces.box import Box from rllab.spaces.discrete import Discrete from rllab.spaces.product import Product from rllab.misc import logger def convert_gym_space(space): if isinstance(space, gym.spaces.Box): return Box(low=space.low, high=space.high) elif isinstance(space, gym.spaces.Discrete): return Discrete(n=space.n) elif isinstance(space, gym.spaces.Tuple): return Product([convert_gym_space(x) for x in space.spaces])