Skip to content

Add MuJoCo Playground environments - #412

Merged
Trinkle23897 merged 13 commits into
mainfrom
jiayi/mujoco-playground-all-envs
May 15, 2026
Merged

Add MuJoCo Playground environments#412
Trinkle23897 merged 13 commits into
mainfrom
jiayi/mujoco-playground-all-envs

Conversation

@Trinkle23897

Copy link
Copy Markdown
Collaborator

Summary

  • add native EnvPool support for MuJoCo Playground v0.2.0 non-DM-Control tasks: 19 locomotion + 10 manipulation envs
  • vendor pinned Playground/Menagerie assets and patches under third_party/mujoco_playground
  • add registry, deterministic, pixel-only, oracle alignment, and render comparison coverage
  • document supported task IDs and add EnvPool-vs-official render comparison image

Notes

  • MuJoCo Playground vendored DM Control Suite tasks are intentionally not registered here; EnvPool already exposes them through the existing dm_control family.

Local validation

  • python3 -m ruff check scripts/render_compare.py envpool/mujoco/playground/playground_oracle_align_test.py envpool/mujoco/playground/playground_test.py envpool/mujoco/playground/registration.py envpool/mujoco/playground/init.py
  • PYTHONPYCACHEPREFIX=/private/tmp/envpool-pycache python3 -m py_compile scripts/render_compare.py envpool/mujoco/playground/playground_oracle_align_test.py envpool/mujoco/playground/playground_test.py envpool/mujoco/playground/registration.py envpool/mujoco/playground/init.py
  • git diff --check
  • bazel test --config=debug //envpool/mujoco:playground_test //envpool/mujoco:playground_pixel_observation_test //envpool/mujoco:playground_oracle_align_test --test_output=errors

@Trinkle23897
Trinkle23897 force-pushed the jiayi/mujoco-playground-all-envs branch 2 times, most recently from e1aaaab to 932268b Compare May 14, 2026 14:32

@chatgpt-codex-connector chatgpt-codex-connector Bot left a comment

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

💡 Codex Review

Here are some automated review suggestions for this pull request.

Reviewed commit: a8c35e66f5

ℹ️ About Codex in GitHub

Your team has set up Codex to review pull requests in this repo. Reviews are triggered when you

  • Open a pull request for review
  • Mark a draft as ready
  • Comment "@codex review".

If Codex has suggestions, it will comment; otherwise it will react with 👍.

Codex can also answer questions or update the PR. Try commenting "@codex address that feedback".

Comment thread scripts/render_compare.py
Comment on lines +42 to +47
if platform.system() != "Linux" or os.environ.get("MUJOCO_GL"):
return

for backend in ("egl", "osmesa"):
probe_env = os.environ.copy()
probe_env["MUJOCO_GL"] = backend

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

P2 Badge Set EGL_PLATFORM when selecting EGL backend

On headless Linux systems that require surfaceless EGL, this helper can mis-detect a usable EGL setup because it probes with MUJOCO_GL=egl but never sets EGL_PLATFORM=surfaceless, and it also exits early when MUJOCO_GL is already set (without normalizing EGL_PLATFORM). In that case the probe may fail, no backend is selected, and later MuJoCo rendering can fail even though EGL would work with the correct platform setting.

Useful? React with 👍 / 👎.

@Trinkle23897
Trinkle23897 force-pushed the jiayi/mujoco-playground-all-envs branch 20 times, most recently from f6fcb21 to cc04d41 Compare May 14, 2026 18:37
@Trinkle23897
Trinkle23897 force-pushed the jiayi/mujoco-playground-all-envs branch from cc04d41 to 18942d4 Compare May 14, 2026 19:17
@Trinkle23897
Trinkle23897 merged commit 671f362 into main May 15, 2026
12 checks passed
@Trinkle23897
Trinkle23897 deleted the jiayi/mujoco-playground-all-envs branch May 15, 2026 20:12
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant