enh: type hinting for bright.py

DC-analysis · Nov 26, 2024 · 525faa2 · 525faa2
1 parent c2fbfbd
commit 525faa2
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/dclab/features/bright.py b/dclab/features/bright.py
@@ -3,9 +3,13 @@
 RT-DC event image mask.
 """
 import numpy as np
+import numpy.typing as npt
 
 
-def get_bright(mask, image, ret_data="avg,sd"):
+def get_bright(
+        mask: npt.NDArray[np.bool] | list[npt.NDArray[np.bool]],
+        image: npt.NDArray | list[npt.NDArray],
+        ret_data: str = "avg,sd"):
     """Compute avg and/or std of the event brightness
 
     The event brightness is defined by the gray-scale values of the