Move DriveMode & TorqueMode

lerobot/common/motors/__init__.py

@@ -1 +1 @@
-from .motors_bus import CalibrationMode, DriveMode, Motor, MotorsBus, TorqueMode
+from .motors_bus import CalibrationMode, Motor, MotorsBus

lerobot/common/motors/dynamixel/__init__.py

@@ -1,3 +1,3 @@
-from .dynamixel import DynamixelMotorsBus, OperatingMode
+from .dynamixel import DriveMode, DynamixelMotorsBus, OperatingMode, TorqueMode
 from .dynamixel_calibration import run_arm_calibration
 from .tables import *

lerobot/common/motors/dynamixel/dynamixel.py

@@ -65,6 +65,16 @@ class OperatingMode(Enum):
     PWM = 16
 
 
+class DriveMode(Enum):
+    NON_INVERTED = 0
+    INVERTED = 1
+
+
+class TorqueMode(Enum):
+    ENABLED = 1
+    DISABLED = 0
+
+
 class DynamixelMotorsBus(MotorsBus):
     """
     The Dynamixel implementation for a MotorsBus. It relies on the python dynamixel sdk to communicate with

lerobot/common/motors/feetech/__init__.py

@@ -1,3 +1,3 @@
-from .feetech import FeetechMotorsBus, OperatingMode
+from .feetech import DriveMode, FeetechMotorsBus, OperatingMode, TorqueMode
 from .feetech_calibration import apply_feetech_offsets_from_calibration, run_full_arm_calibration
 from .tables import *

lerobot/common/motors/feetech/feetech.py

@@ -47,6 +47,16 @@ class OperatingMode(Enum):
     STEP = 3
 
 
+class DriveMode(Enum):
+    NON_INVERTED = 0
+    INVERTED = 1
+
+
+class TorqueMode(Enum):
+    ENABLED = 1
+    DISABLED = 0
+
+
 class FeetechMotorsBus(MotorsBus):
     """
     The FeetechMotorsBus class allows to efficiently read and write to the attached motors. It relies on the

lerobot/common/motors/motors_bus.py

@@ -78,16 +78,6 @@ def assert_same_address(model_ctrl_table: dict[str, dict], motor_models: list[st
         )
 
 
-class TorqueMode(Enum):
-    ENABLED = 1
-    DISABLED = 0
-
-
-class DriveMode(Enum):
-    NON_INVERTED = 0
-    INVERTED = 1
-
-
 class CalibrationMode(Enum):
     DEGREE = 0
     RANGE_0_100 = 1