diff --git a/pyproject.toml b/pyproject.toml index f2be357e..dbc25805 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -166,7 +166,7 @@ lerobot-calibrate="lerobot.scripts.lerobot_calibrate:main" lerobot-find-cameras="lerobot.scripts.lerobot_find_cameras:main" lerobot-find-port="lerobot.scripts.lerobot_find_port:main" lerobot-record="lerobot.scripts.lerobot_record:main" -lerobot-replay="lerobot.replay:main" +lerobot-replay="lerobot.scripts.lerobot_replay:main" lerobot-setup-motors="lerobot.scripts.lerobot_setup_motors:main" lerobot-teleoperate="lerobot.scripts.lerobot_teleoperate:main" lerobot-eval="lerobot.scripts.eval:main" diff --git a/src/lerobot/replay.py b/src/lerobot/scripts/lerobot_replay.py similarity index 100% rename from src/lerobot/replay.py rename to src/lerobot/scripts/lerobot_replay.py diff --git a/tests/test_control_robot.py b/tests/test_control_robot.py index a1dd3328..ace0aea4 100644 --- a/tests/test_control_robot.py +++ b/tests/test_control_robot.py @@ -16,9 +16,9 @@ from unittest.mock import patch -from lerobot.replay import DatasetReplayConfig, ReplayConfig, replay from lerobot.scripts.lerobot_calibrate import CalibrateConfig, calibrate from lerobot.scripts.lerobot_record import DatasetRecordConfig, RecordConfig, record +from lerobot.scripts.lerobot_replay import DatasetReplayConfig, ReplayConfig, replay from lerobot.scripts.lerobot_teleoperate import TeleoperateConfig, teleoperate from tests.fixtures.constants import DUMMY_REPO_ID from tests.mocks.mock_robot import MockRobotConfig