/
githubmirror
/
gym
Обзор
Документация
Войти
/
githubmirror
/
gym
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
master
tests/envs/utils_envs.py
49 строк
1 KB
Mark Towers
Rename "rgb_array" to "rgb_array_list" and "single_rgb_array" to "rgb_array" (#3040)
01 сен 2022, 16:06
Не верифицирован
01 сен 2022, 16:06
799c8d2
Код
Авторство
О чём код?
import gym class RegisterDuringMakeEnv(gym.Env): """Used in `test_registration.py` to check if `env.make` can import and register an env""" def __init__(self): self.action_space = gym.spaces.Discrete(1) self.observation_space = gym.spaces.Discrete(1) class ArgumentEnv(gym.Env): observation_space = gym.spaces.Box(low=-1, high=1, shape=(1,)) action_space = gym.spaces.Box(low=-1, high=1, shape=(1,)) def __init__(self, arg1, arg2, arg3): self.arg1 = arg1 self.arg2 = arg2 self.arg3 = arg3 # Environments to test render_mode class NoHuman(gym.Env): """Environment that does not have human-rendering.""" metadata = {"render_modes": ["rgb_array_list"], "render_fps": 4} def __init__(self, render_mode=None): assert render_mode in self.metadata["render_modes"] self.render_mode = render_mode class NoHumanOldAPI(gym.Env): """Environment that does not have human-rendering.""" metadata = {"render_modes": ["rgb_array_list"], "render_fps": 4} def __init__(self): pass class NoHumanNoRGB(gym.Env): """Environment that has neither human- nor rgb-rendering""" metadata = {"render_modes": ["ascii"], "render_fps": 4} def __init__(self, render_mode=None): assert render_mode in self.metadata["render_modes"] self.render_mode = render_mode