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

Skip tests if alexander koch robot is not available

Browse Source
This commit is contained in:
Remi Cadene
2024-07-10 19:41:57 +02:00
parent c2388c59be
commit 95de4b7454
6 changed files with 55 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
tests/test_control_robot.py
View File

@@ -4,10 +4,11 @@ from lerobot.common.policies.factory import make_policy
from lerobot.common.robot_devices.robots.factory import make_robot
from lerobot.common.utils.utils import init_hydra_config
from lerobot.scripts.control_robot import record_dataset, replay_episode, run_policy, teleoperate
from tests.utils import DEFAULT_CONFIG_PATH, DEVICE
from tests.utils import DEFAULT_CONFIG_PATH, DEVICE, require_koch
def test_teleoperate():
@require_koch
def test_teleoperate(request):
robot = make_robot("koch")
teleoperate(robot, teleop_time_s=1)
teleoperate(robot, fps=30, teleop_time_s=1)
@@ -15,13 +16,13 @@ def test_teleoperate():
del robot
def test_record_dataset_and_replay_episode_and_run_policy(tmpdir):
@require_koch
def test_record_dataset_and_replay_episode_and_run_policy(tmpdir, request):
robot_name = "koch"
env_name = "koch_real"
policy_name = "act_koch_real"
# root = Path(tmpdir)
root = Path("tmp/data")
root = Path(tmpdir)
repo_id = "lerobot/debug"
robot = make_robot(robot_name)