(self, backend: str = "generalized")
| 14 | |
| 15 | class PushT(PipelineEnv): |
| 16 | def __init__(self, backend: str = "generalized"): |
| 17 | # get path of mbd |
| 18 | sys = mjcf.load(f"{mbd.__path__[0]}/assets/pushT.xml") |
| 19 | |
| 20 | super().__init__(sys, backend=backend, n_frames=5) |
| 21 | |
| 22 | def reset(self, rng: jnp.ndarray) -> State: |
| 23 | rng, rng_goal_xy = jax.random.split(rng) |
nothing calls this directly
no outgoing calls
no test coverage detected