From ea9fa69435a8688e0a7a4f2934a131f05477e7f5 Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
<66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Thu, 3 Apr 2025 16:23:42 +0000
Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
---
lerobot/common/robot_devices/control_utils.py | 9 +++++----
1 file changed, 5 insertions(+), 4 deletions(-)
diff --git a/lerobot/common/robot_devices/control_utils.py b/lerobot/common/robot_devices/control_utils.py
index 9ecf77d5..cf3ab44e 100644
--- a/lerobot/common/robot_devices/control_utils.py
+++ b/lerobot/common/robot_devices/control_utils.py
@@ -26,9 +26,9 @@ from functools import cache
import cv2
import torch
+import tqdm
from deepdiff import DeepDiff
from termcolor import colored
-import tqdm
from lerobot.common.datasets.image_writer import safe_stop_image_writer
from lerobot.common.datasets.lerobot_dataset import LeRobotDataset
@@ -300,20 +300,21 @@ def reset_environment(robot, events, reset_time_s, fps):
start_loop_t = time.perf_counter()
robot.teleop_step(record_data=False)
timestamp = time.perf_counter() - start_vencod_t
-
+
# Update progress bar every second
current_second = int(timestamp)
if current_second > last_update:
pbar.update(current_second - last_update)
last_update = current_second
-
+
if events["exit_early"]:
events["exit_early"] = False
break
-
+
dt_s = time.perf_counter() - start_loop_t
busy_wait(1 / fps - dt_s)
+
def stop_recording(robot, listener, display_cameras):
robot.disconnect()