Make ruff linter happy

jackvial · Dec 28, 2024 · 6715331 · 6715331
1 parent 35fa3de
commit 6715331
Show file tree

Hide file tree

Showing 4 changed files with 22 additions and 21 deletions.
diff --git a/lerobot/common/robot_devices/control_context.py b/lerobot/common/robot_devices/control_context.py
@@ -1,14 +1,16 @@
-import numpy as np
+import base64
+import time
+from dataclasses import dataclass
 from typing import Dict, Optional
-from dataclasses import dataclass, asdict
+
 import cv2
-import base64
-import zmq
+import numpy as np
 import torch
-import time
+import zmq
+
+from lerobot.common.robot_devices.control_utils import log_control_info, serialize_log_items
 from lerobot.common.robot_devices.robots.utils import Robot
 from lerobot.common.robot_devices.utils import busy_wait
-from lerobot.common.robot_devices.control_utils import log_control_info, serialize_log_items
 
 
 class ControlPhase:
@@ -240,9 +242,10 @@ def cleanup(self, robot=None):
 
 
 if __name__ == "__main__":
+    import time
+
     import cv2
     import numpy as np
-    import time
     import torch
 
     def read_image_from_camera(cap):

diff --git a/lerobot/common/robot_devices/control_utils.py b/lerobot/common/robot_devices/control_utils.py
@@ -8,9 +8,10 @@
 import traceback
 from contextlib import nullcontext
 from copy import copy
+from dataclasses import asdict, dataclass
 from functools import cache
+from typing import Any, Dict, List, Optional
 
-import cv2
 import torch
 import tqdm
 from deepdiff import DeepDiff
@@ -21,11 +22,9 @@
 from lerobot.common.datasets.utils import get_features_from_robot
 from lerobot.common.policies.factory import make_policy
 from lerobot.common.robot_devices.robots.utils import Robot
-from lerobot.common.robot_devices.utils import busy_wait
 from lerobot.common.utils.utils import get_safe_torch_device, init_hydra_config, set_global_seed
 from lerobot.scripts.eval import get_pretrained_policy_path
-from dataclasses import dataclass, asdict
-from typing import Any, Dict, List, Optional
+
 
 @dataclass
 class LogItem:

diff --git a/lerobot/scripts/browser_ui_server.py b/lerobot/scripts/browser_ui_server.py
@@ -1,10 +1,11 @@
-from flask import Flask, render_template
-from flask_socketio import SocketIO
 import threading
 import time
-import zmq
 from pathlib import Path
 
+import zmq
+from flask import Flask, render_template
+from flask_socketio import SocketIO
+
 # Get template directory path
 template_dir = Path(__file__).resolve().parent.parent / "templates"
 

diff --git a/lerobot/scripts/control_robot.py b/lerobot/scripts/control_robot.py
@@ -101,6 +101,11 @@
 
 # from safetensors.torch import load_file, save_file
 from lerobot.common.datasets.lerobot_dataset import LeRobotDataset
+from lerobot.common.robot_devices.control_context import (
+    ControlContext,
+    ControlContextConfig,
+    ControlPhase,
+)
 from lerobot.common.robot_devices.control_utils import (
     control_loop,
     has_method,
@@ -112,18 +117,11 @@
     sanity_check_dataset_robot_compatibility,
     warmup_record,
 )
-from lerobot.common.robot_devices.control_context import (
-    ControlContext,
-    ControlContextConfig,
-    ControlPhase,
-)
-
 from lerobot.common.robot_devices.robots.factory import make_robot
 from lerobot.common.robot_devices.robots.utils import Robot
 from lerobot.common.robot_devices.utils import busy_wait, safe_disconnect
 from lerobot.common.utils.utils import init_hydra_config, init_logging, log_say, none_or_int
 
-
 ########################################################################################
 # Control modes
 ########################################################################################