robot_type nit

This commit is contained in:
Michel Aractingi
2025-05-07 13:59:21 +02:00
parent 633edcb3af
commit bdd9229576

View File

@@ -1012,7 +1012,7 @@ class EEActionWrapper(gym.ActionWrapper):
self.use_gripper = use_gripper
# Initialize kinematics instance for the appropriate robot type
robot_type = getattr(env.unwrapped.robot.config, "robot_type", "so100")
robot_type = getattr(env.unwrapped.robot.config, "type", "so100")
self.kinematics = RobotKinematics(robot_type)
self.fk_function = self.kinematics.fk_gripper_tip