From af295fadb54dcd5b0601b716dd7bcf10be61b329 Mon Sep 17 00:00:00 2001
From: Simon Alibert <simon.alibert@huggingface.co>
Date: Tue, 25 Mar 2025 12:48:58 +0100
Subject: [PATCH] Fix imports

---
 lerobot/common/motors/dynamixel/dynamixel_calibration.py | 3 ++-
 lerobot/common/motors/feetech/feetech_calibration.py     | 3 +--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/lerobot/common/motors/dynamixel/dynamixel_calibration.py b/lerobot/common/motors/dynamixel/dynamixel_calibration.py
index fbae2e9a..d959aee5 100644
--- a/lerobot/common/motors/dynamixel/dynamixel_calibration.py
+++ b/lerobot/common/motors/dynamixel/dynamixel_calibration.py
@@ -17,7 +17,8 @@
 
 import numpy as np
 
-from ..motors_bus import CalibrationMode, MotorsBus, TorqueMode
+from ..motors_bus import CalibrationMode, MotorsBus
+from .dynamixel import TorqueMode
 from .tables import MODEL_RESOLUTION
 
 URL_TEMPLATE = (
diff --git a/lerobot/common/motors/feetech/feetech_calibration.py b/lerobot/common/motors/feetech/feetech_calibration.py
index 102b30be..098370d0 100644
--- a/lerobot/common/motors/feetech/feetech_calibration.py
+++ b/lerobot/common/motors/feetech/feetech_calibration.py
@@ -16,9 +16,8 @@ import numpy as np
 from ..motors_bus import (
     CalibrationMode,
     MotorsBus,
-    TorqueMode,
 )
-from .feetech import FeetechMotorsBus
+from .feetech import FeetechMotorsBus, TorqueMode
 from .tables import MODEL_RESOLUTION
 
 URL_TEMPLATE = (