tangger/lerobot
1
0
Fork 1
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

[pre-commit.ci] auto fixes from pre-commit.com hooks

Browse Source 
for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot]
2025-05-20 11:47:12 +00:00
parent 3890c38c12
commit 3a08eddeab
1 changed files with 3 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
lerobot/common/cameras/opencv/camera_opencv.py
View File

@@ -33,7 +33,7 @@ from lerobot.common.errors import DeviceAlreadyConnectedError, DeviceNotConnecte
from lerobot.common.utils.utils import capture_timestamp_utc from lerobot.common.utils.utils import capture_timestamp_utc
from ..camera import Camera from ..camera import Camera
from ..utils import IndexOrPath, get_cv2_backend, get_cv2_rotation from ..utils import get_cv2_backend, get_cv2_rotation
from .configuration_opencv import ColorMode, OpenCVCameraConfig from .configuration_opencv import ColorMode, OpenCVCameraConfig
# NOTE(Steven): The maximum opencv device index depends on your operating system. For instance, # NOTE(Steven): The maximum opencv device index depends on your operating system. For instance,
@@ -126,7 +126,6 @@ class OpenCVCamera(Camera):
self.stop_event: Event | None = None self.stop_event: Event | None = None
self.frame_queue: queue.Queue = queue.Queue(maxsize=1) self.frame_queue: queue.Queue = queue.Queue(maxsize=1)
self.rotation: int | None = get_cv2_rotation(config.rotation) self.rotation: int | None = get_cv2_rotation(config.rotation)
self.backend: int = get_cv2_backend() # NOTE(Steven): If we specify backend the opencv open fails self.backend: int = get_cv2_backend() # NOTE(Steven): If we specify backend the opencv open fails
@@ -267,9 +266,7 @@ class OpenCVCamera(Camera):
logger.debug(f"Capture height set to {actual_height} for {self}.") logger.debug(f"Capture height set to {actual_height} for {self}.")
@staticmethod @staticmethod
def find_cameras( def find_cameras(max_index_search_range=MAX_OPENCV_INDEX) -> List[Dict[str, Any]]:
max_index_search_range=MAX_OPENCV_INDEX
) -> List[Dict[str, Any]]:
""" """
Detects available OpenCV cameras connected to the system. Detects available OpenCV cameras connected to the system.
@@ -534,10 +531,7 @@ class OpenCVCamera(Camera):
DeviceNotConnectedError: If the camera is already disconnected. DeviceNotConnectedError: If the camera is already disconnected.
""" """
if not self.is_connected and self.thread is None: if not self.is_connected and self.thread is None:
raise DeviceNotConnectedError( raise DeviceNotConnectedError(f"{self} not connected.")
f"{self} not connected."
)
if self.thread is not None: if self.thread is not None:
self._shutdown_read_thread() self._shutdown_read_thread()