robot_type nit

This commit is contained in:
Michel Aractingi
2025-05-07 13:59:21 +02:00
committed by AdilZouitine
parent f762e2758f
commit 7b45c56be5

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