Fix --width --height type parsing on opencv script

huggingface · Dec 7, 2024 · 6f064ab · 6f064ab
1 parent 8e7d697
commit 6f064ab
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lerobot/common/robot_devices/cameras/opencv.py b/lerobot/common/robot_devices/cameras/opencv.py
@@ -494,13 +494,13 @@ def __del__(self):
     )
     parser.add_argument(
         "--width",
-        type=str,
+        type=int,
         default=None,
         help="Set the width for all cameras. If not provided, use the default width of each camera.",
     )
     parser.add_argument(
         "--height",
-        type=str,
+        type=int,
         default=None,
         help="Set the height for all cameras. If not provided, use the default height of each camera.",
     )