From bf1c737858a68d1141717713e2aff09738594bd3 Mon Sep 17 00:00:00 2001 From: Simon Alibert Date: Thu, 17 Apr 2025 13:18:32 +0200 Subject: [PATCH] Fix calibration msg display --- lerobot/common/robots/koch/koch_follower.py | 2 +- lerobot/common/robots/so100/so100_follower.py | 2 +- lerobot/common/robots/viperx/viperx.py | 2 +- lerobot/common/teleoperators/koch/koch_leader.py | 2 +- lerobot/common/teleoperators/so100/so100_leader.py | 2 +- lerobot/common/teleoperators/widowx/widowx.py | 2 +- 6 files changed, 6 insertions(+), 6 deletions(-) diff --git a/lerobot/common/robots/koch/koch_follower.py b/lerobot/common/robots/koch/koch_follower.py index 2395118d..9572024d 100644 --- a/lerobot/common/robots/koch/koch_follower.py +++ b/lerobot/common/robots/koch/koch_follower.py @@ -122,7 +122,7 @@ class KochFollower(Robot): full_turn_motors = ["shoulder_pan", "wrist_roll"] unknown_range_motors = [name for name in self.arm.names if name not in full_turn_motors] - logger.info( + print( f"Move all joints except {full_turn_motors} sequentially through their entire " "ranges of motion.\nRecording positions. Press ENTER to stop..." ) diff --git a/lerobot/common/robots/so100/so100_follower.py b/lerobot/common/robots/so100/so100_follower.py index 50361fc9..d80e3075 100644 --- a/lerobot/common/robots/so100/so100_follower.py +++ b/lerobot/common/robots/so100/so100_follower.py @@ -121,7 +121,7 @@ class SO100Follower(Robot): full_turn_motor = "wrist_roll" unknown_range_motors = [name for name in self.arm.names if name != full_turn_motor] - logger.info( + print( f"Move all joints except '{full_turn_motor}' sequentially through their " "entire ranges of motion.\nRecording positions. Press ENTER to stop..." ) diff --git a/lerobot/common/robots/viperx/viperx.py b/lerobot/common/robots/viperx/viperx.py index 76287b2d..67877870 100644 --- a/lerobot/common/robots/viperx/viperx.py +++ b/lerobot/common/robots/viperx/viperx.py @@ -117,7 +117,7 @@ class ViperX(Robot): full_turn_motors = ["shoulder_pan", "wrist_roll"] unknown_range_motors = [name for name in self.arm.names if name not in full_turn_motors] - logger.info( + print( f"Move all joints except {full_turn_motors} sequentially through their entire " "ranges of motion.\nRecording positions. Press ENTER to stop..." ) diff --git a/lerobot/common/teleoperators/koch/koch_leader.py b/lerobot/common/teleoperators/koch/koch_leader.py index 85cad16c..4bf33cc8 100644 --- a/lerobot/common/teleoperators/koch/koch_leader.py +++ b/lerobot/common/teleoperators/koch/koch_leader.py @@ -102,7 +102,7 @@ class KochLeader(Teleoperator): full_turn_motors = ["shoulder_pan", "wrist_roll"] unknown_range_motors = [name for name in self.arm.names if name not in full_turn_motors] - logger.info( + print( f"Move all joints except {full_turn_motors} sequentially through their " "entire ranges of motion.\nRecording positions. Press ENTER to stop..." ) diff --git a/lerobot/common/teleoperators/so100/so100_leader.py b/lerobot/common/teleoperators/so100/so100_leader.py index 0ed5eafc..2eeed838 100644 --- a/lerobot/common/teleoperators/so100/so100_leader.py +++ b/lerobot/common/teleoperators/so100/so100_leader.py @@ -96,7 +96,7 @@ class SO100Leader(Teleoperator): full_turn_motor = "wrist_roll" unknown_range_motors = [name for name in self.arm.names if name != full_turn_motor] - logger.info( + print( f"Move all joints except '{full_turn_motor}' sequentially through their " "entire ranges of motion.\nRecording positions. Press ENTER to stop..." ) diff --git a/lerobot/common/teleoperators/widowx/widowx.py b/lerobot/common/teleoperators/widowx/widowx.py index 8b452055..a8c8a7ac 100644 --- a/lerobot/common/teleoperators/widowx/widowx.py +++ b/lerobot/common/teleoperators/widowx/widowx.py @@ -99,7 +99,7 @@ class WidowX(Teleoperator): full_turn_motors = ["shoulder_pan", "wrist_roll"] unknown_range_motors = [name for name in self.arm.names if name not in full_turn_motors] - logger.info( + print( f"Move all joints except {full_turn_motors} sequentially through their " "entire ranges of motion.\nRecording positions. Press ENTER to stop..." )