From 6ccf083127424cf7fbbdb0ed2321dd57528378d4 Mon Sep 17 00:00:00 2001 From: Simon Alibert Date: Tue, 25 Mar 2025 12:32:38 +0100 Subject: [PATCH] Update so100 imports --- lerobot/common/robots/so100/robot_so100.py | 3 ++- lerobot/common/teleoperators/so100/teleop_so100.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/lerobot/common/robots/so100/robot_so100.py b/lerobot/common/robots/so100/robot_so100.py index ccbe9dad..a0321477 100644 --- a/lerobot/common/robots/so100/robot_so100.py +++ b/lerobot/common/robots/so100/robot_so100.py @@ -22,10 +22,11 @@ import numpy as np from lerobot.common.cameras.utils import make_cameras_from_configs from lerobot.common.constants import OBS_IMAGES, OBS_STATE from lerobot.common.errors import DeviceAlreadyConnectedError, DeviceNotConnectedError -from lerobot.common.motors import CalibrationMode, Motor, TorqueMode +from lerobot.common.motors import CalibrationMode, Motor from lerobot.common.motors.feetech import ( FeetechMotorsBus, OperatingMode, + TorqueMode, apply_feetech_offsets_from_calibration, ) diff --git a/lerobot/common/teleoperators/so100/teleop_so100.py b/lerobot/common/teleoperators/so100/teleop_so100.py index 0e02b313..ab0bf3c0 100644 --- a/lerobot/common/teleoperators/so100/teleop_so100.py +++ b/lerobot/common/teleoperators/so100/teleop_so100.py @@ -20,9 +20,10 @@ import time import numpy as np from lerobot.common.errors import DeviceAlreadyConnectedError, DeviceNotConnectedError -from lerobot.common.motors import CalibrationMode, Motor, TorqueMode +from lerobot.common.motors import CalibrationMode, Motor from lerobot.common.motors.feetech import ( FeetechMotorsBus, + TorqueMode, apply_feetech_offsets_from_calibration, )