Skip to content

Commit 1c7c709

Browse files
Remove Mujoco assert on metadata render-mode (#1328)
1 parent 583f6a7 commit 1c7c709

File tree

2 files changed

+30
-6
lines changed

2 files changed

+30
-6
lines changed

Diff for: gymnasium/envs/mujoco/mujoco_env.py

-6
Original file line numberDiff line numberDiff line change
@@ -82,12 +82,6 @@ def __init__(
8282

8383
self.frame_skip = frame_skip
8484

85-
assert self.metadata["render_modes"] == [
86-
"human",
87-
"rgb_array",
88-
"depth_array",
89-
"rgbd_tuple",
90-
], self.metadata["render_modes"]
9185
if "render_fps" in self.metadata:
9286
assert (
9387
int(np.round(1.0 / self.dt)) == self.metadata["render_fps"]

Diff for: tests/envs/mujoco/test_mujoco_rendering.py

+30
Original file line numberDiff line numberDiff line change
@@ -60,6 +60,36 @@ def test_offscreen_viewer_custom_dimensions(
6060
viewer.close()
6161

6262

63+
@pytest.mark.parametrize(
64+
"env_id",
65+
[
66+
"Ant",
67+
"HalfCheetah",
68+
"Hopper",
69+
"Humanoid",
70+
"HumanoidStandup",
71+
"InvertedDoublePendulum",
72+
"InvertedPendulum",
73+
"Pusher",
74+
"Reacher",
75+
"Swimmer",
76+
"Walker2d",
77+
],
78+
)
79+
@pytest.mark.parametrize("version", ["v4", "v5"])
80+
def test_mujoco_metadata_render_modes(env_id, version):
81+
if env_id == "Pusher" and version == "v4":
82+
pytest.skip()
83+
env = gymnasium.make(f"{env_id}-{version}")
84+
assert env.metadata["render_modes"] == [
85+
"human",
86+
"rgb_array",
87+
"depth_array",
88+
"rgbd_tuple",
89+
]
90+
env.close()
91+
92+
6393
@pytest.mark.parametrize(
6494
"render_mode", ["human", "rgb_array", "depth_array", "rgbd_tuple"]
6595
)

0 commit comments

Comments
 (0)