wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix(motors): Allows motor to be called other than gripper

This commit is contained in:
Steven Palma 2025-03-17 18:16:52 +01:00
parent f75c9dbdda
commit 6d42af98b6
No known key found for this signature in database
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lerobot/common/motors/feetech/feetech_calibration.py
View File

@ -33,7 +33,7 @@ def disable_torque(arm: MotorsBus):
def get_calibration_modes(arm: MotorsBus): def get_calibration_modes(arm: MotorsBus):
"""Returns calibration modes for each motor (DEGREE for rotational, LINEAR for gripper).""" """Returns calibration modes for each motor (DEGREE for rotational, LINEAR for gripper)."""
return [ return [
CalibrationMode.LINEAR.name if name == "gripper" else CalibrationMode.DEGREE.name CalibrationMode.LINEAR.name if "gripper" in name else CalibrationMode.DEGREE.name
for name in arm.motor_names for name in arm.motor_names
] ]
@ -217,7 +217,7 @@ def apply_feetech_offsets_from_calibration(motorsbus: FeetechMotorsBus, calibrat
print(f"Warning: '{m_name}' not found in motorsbus.motors; skipping offset.") print(f"Warning: '{m_name}' not found in motorsbus.motors; skipping offset.")
continue continue
if m_name == "gripper": if "gripper" in m_name:
old_offset = start_pos # If gripper set the offset to the start position of the gripper old_offset = start_pos # If gripper set the offset to the start position of the gripper
continue continue