diff --git a/lerobot/scripts/server/gym_manipulator.py b/lerobot/scripts/server/gym_manipulator.py index 1f3f78202..7608cc9d6 100644 --- a/lerobot/scripts/server/gym_manipulator.py +++ b/lerobot/scripts/server/gym_manipulator.py @@ -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