From e0394961cbb02709e08b9a9159f8b7165bb56edc Mon Sep 17 00:00:00 2001 From: Remi Cadene Date: Mon, 14 Oct 2024 17:08:08 +0200 Subject: [PATCH] small fix unit tests --- tests/test_control_robot.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/test_control_robot.py b/tests/test_control_robot.py index 19a83780..4df858b0 100644 --- a/tests/test_control_robot.py +++ b/tests/test_control_robot.py @@ -319,7 +319,7 @@ def test_record_with_event_rerecord_episode(tmpdir, request, robot_type, mock): robot = make_robot(robot_type, overrides=overrides, mock=mock) with ( patch("lerobot.scripts.control_robot.init_keyboard_listener") as mock_listener, - patch("lerobot.common.robot_devices.control_robot.add_frame", wraps=add_frame) as mock_add_frame, + patch("lerobot.common.robot_devices.control_utils.add_frame", wraps=add_frame) as mock_add_frame, ): mock_events = {} mock_events["exit_early"] = True @@ -368,7 +368,7 @@ def test_record_with_event_exit_early(tmpdir, request, robot_type, mock): robot = make_robot(robot_type, overrides=overrides, mock=mock) with ( patch("lerobot.scripts.control_robot.init_keyboard_listener") as mock_listener, - patch("lerobot.common.robot_devices.control_robot.add_frame", wraps=add_frame) as mock_add_frame, + patch("lerobot.common.robot_devices.control_utils.add_frame", wraps=add_frame) as mock_add_frame, ): mock_events = {} mock_events["exit_early"] = True @@ -418,7 +418,7 @@ def test_record_with_event_stop_recording(tmpdir, request, robot_type, mock, num robot = make_robot(robot_type, overrides=overrides, mock=mock) with ( patch("lerobot.scripts.control_robot.init_keyboard_listener") as mock_listener, - patch("lerobot.common.robot_devices.control_robot.add_frame", wraps=add_frame) as mock_add_frame, + patch("lerobot.common.robot_devices.control_utils.add_frame", wraps=add_frame) as mock_add_frame, ): mock_events = {} mock_events["exit_early"] = True