diff --git a/pyproject.toml b/pyproject.toml index 36746e79..54458486 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 708340ab..89d5995f 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)