nit

huggingface · Oct 9, 2024 · 451d50e · 451d50e
1 parent 874dfc4
commit 451d50e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/lerobot/common/robot_devices/robots/manipulator.py b/lerobot/common/robot_devices/robots/manipulator.py
@@ -364,6 +364,7 @@ def connect(self):
         for name in self.follower_arms:
             print(f"Connecting {name} follower arm.")
             self.follower_arms[name].connect()
+        for name in self.leader_arms:
             print(f"Connecting {name} leader arm.")
             self.leader_arms[name].connect()
 

diff --git a/tests/test_control_robot.py b/tests/test_control_robot.py
@@ -110,7 +110,7 @@ def test_record_without_cameras(tmpdir, request, robot_type, mock):
 @require_robot
 def test_record_and_replay_and_policy(tmpdir, request, robot_type, mock):
     if mock:
-        # TODO(rcadene, aliberts): hack DO NOT MERGE
+        # Avoid hanging issue when running two consecutive tests
         multiprocessing.set_start_method("spawn", force=True)
 
         request.getfixturevalue("patch_builtins_input")