Without this fix, the test on the installed (in a non-editable way) brax package fail with:
2023-04-26T10:15:12.5536862Z FAILED io/mjcf_test.py::MjcfTest::test_load_urdf - FileNotFoundError: [Errno 2] No such file or directory: '$PREFIX/lib/python3.10/site-packages/brax/envs/assets/laikago/laikago_toes_zup.urdf'
Without this fix, the test on the installed (in a non-editable way) brax package fail with:
error.
Similar to https://github.com/google/brax/pull/292/files .