From 91f549b2cecab2849f645e35b1d3c050fa9783d0 Mon Sep 17 00:00:00 2001 From: Pepijn Date: Tue, 18 Mar 2025 13:59:50 +0100 Subject: [PATCH] use float32 instead of int --- lerobot/common/robot_devices/robots/manipulator.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/lerobot/common/robot_devices/robots/manipulator.py b/lerobot/common/robot_devices/robots/manipulator.py index 2c358cb9..7b7bb36d 100644 --- a/lerobot/common/robot_devices/robots/manipulator.py +++ b/lerobot/common/robot_devices/robots/manipulator.py @@ -504,7 +504,7 @@ class ManipulatorRobot: # Used when record_data=True follower_goal_pos[name] = goal_pos - goal_pos = goal_pos.numpy().astype(np.int32) + goal_pos = goal_pos.numpy().astype(np.float32) self.follower_arms[name].write("Goal_Position", goal_pos) self.logs[f"write_follower_{name}_goal_pos_dt_s"] = time.perf_counter() - before_fwrite_t @@ -626,7 +626,7 @@ class ManipulatorRobot: action_sent.append(goal_pos) # Send goal position to each follower - goal_pos = goal_pos.numpy().astype(np.int32) + goal_pos = goal_pos.numpy().astype(np.float32) self.follower_arms[name].write("Goal_Position", goal_pos) return torch.cat(action_sent)