Skip to content

Commit 3cc2f56

Browse files
skip explicit torso action is >1D
1 parent 69330f1 commit 3cc2f56

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

robosuite/devices/device.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -161,7 +161,9 @@ def input2action(self, mirror_actions=False, goal_update_mode="target") -> Optio
161161
device_torso_input = 0.0 # No torso input for non-mobile robots by default
162162

163163
if hasattr(robot, "torso") and robot.torso is not None:
164-
ac_dict["torso"] = self.get_torso_action(robot, device_torso_input)
164+
# only support single joint torso for now
165+
if self.env.robots[0].composite_controller.part_controllers['torso'].joint_dim == 1:
166+
ac_dict["torso"] = self.get_torso_action(robot, device_torso_input)
165167

166168
# populate action dict items for arm and grippers
167169
arm_action = self.get_arm_action(

0 commit comments

Comments
 (0)