google / brax

Massively parallel rigidbody physics simulation on accelerator hardware.
Apache License 2.0
2.14k stars 234 forks source link

MjxEnv no where to be found #451

Closed Engineering-Geek closed 4 months ago

Engineering-Geek commented 4 months ago

image As shown in the screen shot, MjxEnv is no longer detected anywhere in the code. I swear it was working earlier today on my PC... When I open the file and looked through the code and current main branch, I can't find MjxEnv anywhere now unless if I am blind.

btaba commented 4 months ago

Hi @Engineering-Geek we just did a release and removed MjxEnv. You shouldn't need it anymore! If you'd like to work off of an older version of the code base, you can still use 0.9.4 via pip install

See https://github.com/google/brax/releases/tag/v0.10.0

btaba commented 4 months ago

For more background info, you can use PipelineEnv now and add backend='mjx', and use the MJX backend like any other backend in Brax. Let us know if you have any issues!