Skip to content

Forward **kwargs through Wrapper.render to the wrapped env - #327

Open
JJJEEERRR wants to merge 1 commit into
google-deepmind:mainfrom
JJJEEERRR:fix/wrapper-render-kwargs
Open

Forward **kwargs through Wrapper.render to the wrapped env#327
JJJEEERRR wants to merge 1 commit into
google-deepmind:mainfrom
JJJEEERRR:fix/wrapper-render-kwargs

Conversation

@JJJEEERRR

Copy link
Copy Markdown

Summary

Wrapper.render (the base wrapper in mujoco_playground/_src/wrapper.py)
forwarded only a fixed set of positional arguments to self.env.render, so any
extra render keyword arguments defined by a custom environment were silently
dropped. This makes such kwargs reach the wrapped environment.

Fixes #258.

What changed

  • Wrapper.render now accepts and forwards **kwargs to self.env.render.
  • Forwarding switched to keyword arguments for the existing named parameters,
    for clarity at the call site.
  • Added WrapperTest.test_render_forwards_kwargs, a small regression test using
    a recording env that asserts extra kwargs are passed through.

Backward compatibility

Fully backward compatible: existing call sites pass the same named arguments,
and envs whose render does not accept extra kwargs are unaffected because no
extra kwargs are added by existing callers.

The base `Wrapper.render` forwarded only a fixed set of positional
arguments to `self.env.render`, silently dropping any extra render
kwargs that custom environments define. This adds a `**kwargs`
passthrough (and switches to keyword forwarding for clarity), so
wrapped envs receive all render arguments.

Adds a regression test that a wrapped env receives forwarded kwargs.

Closes google-deepmind#258
@google-cla

google-cla Bot commented Jun 15, 2026

Copy link
Copy Markdown

Thanks for your pull request! It looks like this may be your first contribution to a Google open source project. Before we can look at your pull request, you'll need to sign a Contributor License Agreement (CLA).

View this failed invocation of the CLA check for more information.

For the most up to date status, view the checks section at the bottom of the pull request.

@JJJEEERRR

Copy link
Copy Markdown
Author

@googlebot I signed it!

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.

Allow mujoco_playground._src.wrapper.Wrapper.render to accept kwargs

1 participant