From 9db62138952b6a10e586c56dfffabd1ed407de1b Mon Sep 17 00:00:00 2001 From: Steven Palma Date: Mon, 20 Oct 2025 16:25:03 +0200 Subject: [PATCH] chore(style): update mypy config (#2257) * chore(style): update mypy config * fix(cameras): mypy check --- pyproject.toml | 11 +++-------- src/lerobot/cameras/opencv/camera_opencv.py | 4 ++++ 2 files changed, 7 insertions(+), 8 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index 36746e79c..544584867 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -298,11 +298,6 @@ disallow_untyped_defs = true disallow_incomplete_defs = true check_untyped_defs = true -# ignores -disable_error_code = ["attr-defined"] #TODO: draccus issue - -# include = "src/lerobot/configs/**/*.py" - # [[tool.mypy.overrides]] # module = "lerobot.optim.*" # ignore_errors = false @@ -319,9 +314,9 @@ ignore_errors = false # module = "lerobot.datasets.*" # ignore_errors = false -# [[tool.mypy.overrides]] -# module = "lerobot.cameras.*" -# ignore_errors = false +[[tool.mypy.overrides]] +module = "lerobot.cameras.*" +ignore_errors = false # [[tool.mypy.overrides]] # module = "lerobot.motors.*" diff --git a/src/lerobot/cameras/opencv/camera_opencv.py b/src/lerobot/cameras/opencv/camera_opencv.py index 708340ab8..89d5995f4 100644 --- a/src/lerobot/cameras/opencv/camera_opencv.py +++ b/src/lerobot/cameras/opencv/camera_opencv.py @@ -244,6 +244,10 @@ class OpenCVCamera(Camera): """Validates and sets the camera's FOURCC code.""" fourcc_code = cv2.VideoWriter_fourcc(*self.config.fourcc) + + if self.videocapture is None: + raise DeviceNotConnectedError(f"{self} videocapture is not initialized") + success = self.videocapture.set(cv2.CAP_PROP_FOURCC, fourcc_code) actual_fourcc_code = self.videocapture.get(cv2.CAP_PROP_FOURCC)