mirror of
https://github.com/Farama-Foundation/Gymnasium.git
synced 2025-08-17 20:39:12 +00:00
799c8d20c1
* initial commit * Fix CI * Fixed CI * third time the charm * Fix mujoco environment render modes order * Remove unnecessary changes * pre-commit * Fix tests * Comment out test render modes * Fix code review and readd mujoco * pre-commit * Fix testing envs * Fix all GenericTestEnvs * Do not run mujoco-py render environments
50 lines
1.3 KiB
Python
50 lines
1.3 KiB
Python
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
|