MCPcopy Create free account
hub / github.com/LeCAR-Lab/model-based-diffusion / __init__

Method __init__

mbd/envs/walker2d.py:13–19  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

11class Walker2d(PipelineEnv):
12
13 def __init__(self):
14 path = epath.resource_path("brax") / "envs/assets/walker2d.xml"
15 sys = mjcf.load(path)
16
17 self._reset_noise_scale = 5e-3
18
19 super().__init__(sys=sys, backend="positional", n_frames=20)
20
21 def reset(self, rng: jax.Array) -> State:
22 """Resets the environment to an initial state."""

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected