tangger/lerobot
1
0
Fork 1
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix unit test

Browse Source
This commit is contained in:
Remi Cadene
2024-09-27 12:29:58 +02:00
parent 81f17d505e
commit e499d60742
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/test_control_robot.py
View File

@@ -37,6 +37,9 @@ from tests.utils import DEFAULT_CONFIG_PATH, DEVICE, TEST_ROBOT_TYPES, require_r
@pytest.mark.parametrize("robot_type, mock", TEST_ROBOT_TYPES) @pytest.mark.parametrize("robot_type, mock", TEST_ROBOT_TYPES)
@require_robot @require_robot
def test_teleoperate(request, robot_type, mock): def test_teleoperate(request, robot_type, mock):
if mock:
request.getfixturevalue("patch_builtins_input")
robot = make_robot(robot_type, mock=mock) robot = make_robot(robot_type, mock=mock)
teleoperate(robot, teleop_time_s=1) teleoperate(robot, teleop_time_s=1)
teleoperate(robot, fps=30, teleop_time_s=1) teleoperate(robot, fps=30, teleop_time_s=1)
@@ -85,6 +88,9 @@ def test_record_and_replay_and_policy(tmpdir, request, robot_type, mock):
if mock: if mock:
request.getfixturevalue("patch_builtins_input") request.getfixturevalue("patch_builtins_input")
if robot_type == "aloha":
pytest.skip("TODO(rcadene): enable test once aloha_real and act_aloha_real are merged")
env_name = "koch_real" env_name = "koch_real"
policy_name = "act_koch_real" policy_name = "act_koch_real"