diff --git a/lerobot/common/robot_devices/control_utils.py b/lerobot/common/robot_devices/control_utils.py index e4ed73f7..08bcec2e 100644 --- a/lerobot/common/robot_devices/control_utils.py +++ b/lerobot/common/robot_devices/control_utils.py @@ -224,7 +224,7 @@ def record_episode( @safe_stop_image_writer def control_loop( robot, - control_time_s, + control_time_s=None, teleoperate=False, display_cameras=False, dataset=None, @@ -241,6 +241,9 @@ def control_loop( if events is None: events = {"exit_early": False} + if control_time_s is None: + control_time_s = float("inf") + if teleoperate and policy is not None: raise ValueError("When `teleoperate` is True, `policy` should be None.")