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

reset_time_s=1 in tests

This commit is contained in:
Remi Cadene 2024-10-13 16:27:14 +02:00
parent 63bd5013bf
commit 517a261d71
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_control_robot.py
View File

@ -140,6 +140,7 @@ def test_record_and_replay_and_policy(tmpdir, request, robot_type, mock):
repo_id=repo_id, repo_id=repo_id,
warmup_time_s=1, warmup_time_s=1,
episode_time_s=1, episode_time_s=1,
reset_time_s=1,
num_episodes=2, num_episodes=2,
push_to_hub=False, push_to_hub=False,
# TODO(rcadene, aliberts): test video=True # TODO(rcadene, aliberts): test video=True
@ -214,6 +215,7 @@ def test_record_and_replay_and_policy(tmpdir, request, robot_type, mock):
pretrained_policy_name_or_path, pretrained_policy_name_or_path,
warmup_time_s=1, warmup_time_s=1,
episode_time_s=1, episode_time_s=1,
reset_time_s=1,
num_episodes=2, num_episodes=2,
run_compute_stats=False, run_compute_stats=False,
push_to_hub=False, push_to_hub=False,