diff --git a/tests/test_control_robot.py b/tests/test_control_robot.py index 00683cb9..97379766 100644 --- 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,