fix teleop

This commit is contained in:
Remi Cadene
2024-10-16 20:36:57 +02:00
parent eeea3e5d1d
commit 79ac1ad271

View File

@@ -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.")