reset_time_s=1 in tests

2024-10-13 16:27:14 +02:00 · 2024-10-13 16:27:14 +02:00 · 517a261d71
parent 63bd5013bf
commit 517a261d71
1 changed files with 2 additions and 0 deletions
--- a/tests/test_control_robot.py
+++ b/tests/test_control_robot.py
@ -140,6 +140,7 @@ def test_record_and_replay_and_policy(tmpdir, request, robot_type, mock):
        repo_id=repo_id,
        warmup_time_s=1,
        episode_time_s=1,
+        reset_time_s=1,
        num_episodes=2,
        push_to_hub=False,
        # 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,
        warmup_time_s=1,
        episode_time_s=1,
+        reset_time_s=1,
        num_episodes=2,
        run_compute_stats=False,
        push_to_hub=False,