diff --git a/lerobot/common/motors/dynamixel/dynamixel.py b/lerobot/common/motors/dynamixel/dynamixel.py index b0c9844f..5992ae56 100644 --- a/lerobot/common/motors/dynamixel/dynamixel.py +++ b/lerobot/common/motors/dynamixel/dynamixel.py @@ -57,6 +57,7 @@ class DynamixelMotorsBus(MotorsBus): self.packet_handler = dxl.PacketHandler(PROTOCOL_VERSION) self.sync_reader = dxl.GroupSyncRead(self.port_handler, self.packet_handler, 0, 0) self.sync_writer = dxl.GroupSyncWrite(self.port_handler, self.packet_handler, 0, 0) + self.__comm_success = dxl.COMM_SUCCESS def broadcast_ping( self, num_retry: int = 0, raise_on_error: bool = False @@ -78,9 +79,7 @@ class DynamixelMotorsBus(MotorsBus): return ids_values def _is_comm_success(self, comm: int) -> bool: - import dynamixel_sdk as dxl - - return comm == dxl.COMM_SUCCESS + return comm == self.__comm_success def _is_error(self, error: int) -> bool: return error != 0x00 diff --git a/lerobot/common/motors/feetech/feetech.py b/lerobot/common/motors/feetech/feetech.py index 5d1fa69e..4c8dd4ed 100644 --- a/lerobot/common/motors/feetech/feetech.py +++ b/lerobot/common/motors/feetech/feetech.py @@ -53,6 +53,7 @@ class FeetechMotorsBus(MotorsBus): self.packet_handler = scs.PacketHandler(PROTOCOL_VERSION) self.sync_reader = scs.GroupSyncRead(self.port_handler, self.packet_handler, 0, 0) self.sync_writer = scs.GroupSyncWrite(self.port_handler, self.packet_handler, 0, 0) + self.__comm_success = scs.COMM_SUCCESS def broadcast_ping(self, num_retry: int | None = None): raise NotImplementedError # TODO @@ -66,9 +67,7 @@ class FeetechMotorsBus(MotorsBus): return ids_values def _is_comm_success(self, comm: int) -> bool: - import scservo_sdk as scs - - return comm == scs.COMM_SUCCESS + return comm == self.__comm_success def _is_error(self, error: int) -> bool: return error != 0x00