ENH: allow metadata["render_modes"] to be a set (#248)

This commit is contained in:
Omar Younis
2023-01-08 20:56:52 +01:00
committed by GitHub
parent 8b371450a1
commit 3413b04534
57 changed files with 103 additions and 103 deletions

View File

@@ -100,7 +100,7 @@ from gymnasium import spaces
class GridWorldEnv(gym.Env):
metadata = {"render_modes": ["human", "rgb_array"], "render_fps": 4}
metadata = {"render_modes": {"human", "rgb_array"}, "render_fps": 4}
def __init__(self, render_mode=None, size=5):
self.size = size # The size of the square grid
@@ -130,7 +130,6 @@ class GridWorldEnv(gym.Env):
3: np.array([0, -1]),
}
assert render_mode is None or render_mode in self.metadata["render_modes"]
self.render_mode = render_mode
"""