set_state() not working in walker

[SilviaSapora]

Hello,

I believe the set_state() function does not work for the walker environment.
The walker environment (https://github.com/deepmind/dm_control/blob/main/dm_control/suite/walker.py) defines its own Physics class, that extends the mujoco.Physics class, but does not extend the set_state() function.
I tried implementing the extra functions needed on my own but I believe I would need to build the repo, and that is not possible.
Am I missing something?
If not, can the extra functionality be added?