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, )