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

Method _get_obs

mbd/envs/cartpole.py:54–56  ·  view source on GitHub ↗

Observe cartpole body position and velocities.

(self, pipeline_state: base.State)

Source from the content-addressed store, hash-verified

52 return 1
53
54 def _get_obs(self, pipeline_state: base.State) -> jax.Array:
55 """Observe cartpole body position and velocities."""
56 return jp.concatenate([pipeline_state.q, pipeline_state.qd])

Callers 2

resetMethod · 0.95
stepMethod · 0.95

Calls

no outgoing calls

Tested by

no test coverage detected