Skip to content

Commit b8ea543

Browse files
committed
feat: added tau ext to sim robot
1 parent ba3efe7 commit b8ea543

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

python/rcs/envs/sim.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -65,6 +65,7 @@ def step(self, action: dict[str, Any]) -> tuple[dict[str, Any], float, bool, boo
6565
info["collision"] = info["collision"] or state.collision
6666
info["ik_success"] = state.ik_success
6767
info["is_sim_converged"] = self.sim.is_converged()
68+
obs["tau_ext"] = self.sim.data.qfrc_constraint[:7]
6869
# truncate episode if collision
6970
obs.update(self.unwrapped.get_obs())
7071
return obs, 0, False, info["collision"] or not state.ik_success, info

0 commit comments

Comments
 (0)