wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Speedup test

This commit is contained in:
Simon Alibert 2024-10-26 10:59:06 +02:00
parent e210d795de
commit df3d2ec5df
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tests/test_control_robot.py
View File

@ -140,7 +140,7 @@ def test_record_and_replay_and_policy(tmpdir, request, robot_type, mock):
root, root,
repo_id, repo_id,
single_task, single_task,
fps=5, fps=1,
warmup_time_s=0.5, warmup_time_s=0.5,
episode_time_s=1, episode_time_s=1,
reset_time_s=1, reset_time_s=1,
@ -153,9 +153,9 @@ def test_record_and_replay_and_policy(tmpdir, request, robot_type, mock):
play_sounds=False, play_sounds=False,
) )
assert dataset.total_episodes == 2 assert dataset.total_episodes == 2
assert len(dataset) == 10 assert len(dataset) == 2
replay(robot, episode=0, fps=5, root=root, repo_id=repo_id, play_sounds=False) replay(robot, episode=0, fps=1, root=root, repo_id=repo_id, play_sounds=False)
# TODO(rcadene, aliberts): rethink this design # TODO(rcadene, aliberts): rethink this design
if robot_type == "aloha": if robot_type == "aloha":