Commit 6904649
authored
fix(managers): ManagerBasedRVEnv.reset accepts seed (honor env.reset(seed=) contract) (#800)
ManagerBasedRVEnv (the manager-based RL base for mjlab v2 / Isaac-Lab-style
tasks) overrode reset(self, env_ids=None) without seed, so gym.make(task).reset(seed=)
raised 'TypeError: reset() got an unexpected keyword argument seed' for every
manager-based task — surfaced by the MuJoCo benchmark (12 mjlab v2 tasks failed
to reset). Add seed=None and forward to handler.set_seed (this base reimplements
reset and doesn't call super), matching the seed-contract fix already applied to
the other reimplementing bases (humanoid LeggedRobotTask, beyondmimic).
Verified on MuJoCo (8xRTX5090): mjlab.cartpole_swingup_v2 / cartpole_balance_v2 /
velocity_flat_g1_v2 now load + reset(seed=0) + run a rollout (previously TypeError).1 parent 78587cf commit 6904649
1 file changed
Lines changed: 12 additions & 2 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
336 | 336 | | |
337 | 337 | | |
338 | 338 | | |
339 | | - | |
340 | | - | |
| 339 | + | |
| 340 | + | |
| 341 | + | |
| 342 | + | |
| 343 | + | |
| 344 | + | |
| 345 | + | |
| 346 | + | |
| 347 | + | |
| 348 | + | |
| 349 | + | |
| 350 | + | |
341 | 351 | | |
342 | 352 | | |
343 | 353 | | |
| |||
0 commit comments