Add type annotation for function get_capture_array()

ci-group · Oct 7, 2024 · eb27e06 · eb27e06
1 parent 9b0913c
commit eb27e06
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/robohatlib/hal/Camera.py b/robohatlib/hal/Camera.py
@@ -12,6 +12,7 @@
 except ImportError:
     raise ImportError("Failed to import needed dependencies for the Camera class, make sure picamera2 is installed")
 
+from numpy.typing import NDArray
     # --------------------------------------------------------------------------------------
 
 class Camera:
@@ -52,7 +53,7 @@ def init_camera(self) -> None:
 
     # --------------------------------------------------------------------------------------
 
-    def get_capture_array(self):
+    def get_capture_array(self) -> NDArray[np.uint8]:
         """!
         return numpy arrays of captured data. Makes a copt before leaving
         @return: None or array
@@ -138,4 +139,4 @@ def __check_ffmpeg() -> bool:
             return False
         return True
 
-    # --------------------------------------------------------------------------------------
+    # --------------------------------------------------------------------------------------