google/brax

[MJX] Model loading errors in `brax.io`, for features supported by MJX

varadVaidya opened this issue · 1 comments

With the MuJoCo 3.1.2 update, the site transmission was added to MJX, and a new pipeline to train RL agents using PipelineEnv was added brax. However while loading the model, through this pipeline, errors regarding RK4 integration scheme, and site transmission, are thrown, even though the pipeline is set to mjx.

  • Include the following context:
    • Operating system : Ubuntu 20.04
    • MuJoCo version (and if the bug is new, the version where it used to work): 3.1.2
    • mujoco python bindings

Solved in #1422