small fix unit tests

This commit is contained in:
Remi Cadene
2024-10-14 17:19:25 +02:00
parent e0394961cb
commit cde4945b29
3 changed files with 6 additions and 2 deletions

View File

@@ -316,7 +316,8 @@ def stop_recording(robot, listener, display_cameras):
robot.disconnect()
if not is_headless():
listener.stop()
if listener is not None:
listener.stop()
if display_cameras:
cv2.destroyAllWindows()

View File

@@ -289,6 +289,9 @@ def record(
# Increment by one dataset["current_episode_index"]
save_current_episode(dataset)
if events["stop_recording"]:
break
log_say("Stop recording", play_sounds, blocking=True)
stop_recording(robot, listener, display_cameras)