We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent ba3efe7 commit b8ea543Copy full SHA for b8ea543
1 file changed
python/rcs/envs/sim.py
@@ -65,6 +65,7 @@ def step(self, action: dict[str, Any]) -> tuple[dict[str, Any], float, bool, boo
65
info["collision"] = info["collision"] or state.collision
66
info["ik_success"] = state.ik_success
67
info["is_sim_converged"] = self.sim.is_converged()
68
+ obs["tau_ext"] = self.sim.data.qfrc_constraint[:7]
69
# truncate episode if collision
70
obs.update(self.unwrapped.get_obs())
71
return obs, 0, False, info["collision"] or not state.ik_success, info
0 commit comments