Problem implementing MJX environment from xml (NotImplementedError) about mujoco

[btaba]

Hi @rs545837 what's the full error you're getting with traceback? Can you also paste which version of MuJoCo/MJX you're using?

[rs545837]

Hey @btaba, it's
mujoco==3.0.0
mujoco_mjx==3.0.0

Traceback (most recent call last):
  File "/Users/rohansharma03/Desktop/SCALER_Mujoco/SCALER_XML/RL_Training/file2.py", line 455, in <module>
    env = envs.get_environment(env_name)
          ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/opt/homebrew/lib/python3.11/site-packages/brax/envs/__init__.py", line 63, in get_environment
    return _envs[env_name](**kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/Users/rohansharma03/Desktop/SCALER_Mujoco/SCALER_XML/RL_Training/file2.py", line 120, in __init__
    super().__init__(mj_model=mj_model, **kwargs)
  File "/Users/rohansharma03/Desktop/SCALER_Mujoco/SCALER_XML/RL_Training/file1.py", line 70, in __init__
    self.sys = mjx.device_put(mj_model)
               ^^^^^^^^^^^^^^^^^^^^^^^^
  File "/opt/homebrew/lib/python3.11/site-packages/mujoco/mjx/_src/device.py", line 186, in device_put
    _validate(value)  # type: ignore
    ^^^^^^^^^^^^^^^^
  File "/opt/homebrew/lib/python3.11/site-packages/mujoco/mjx/_src/device.py", line 120, in _validate
    raise NotImplementedError(f'{unsupported_typs} not implemented.')
NotImplementedError: {2} not implemented.

[btaba]

Hey @rs545837 , can you try the latest versions 3.1.1. I suspect the issue may be fixed with the latest versions

[rs545837]

Yup, thanks @btaba, it seems it got resolved in the latest version. My bad that I forgot about the version update.