fix: formatting and structure

Author: jkbmrz

depthai_nodes/ml/messages/creators/__init__.py

@@ -2,10 +2,10 @@
 from .detection import create_detection_message, create_line_detection_message
 from .image import create_image_message
 from .keypoints import create_hand_keypoints_message, create_keypoints_message
+from .misc import create_age_gender_message
 from .segmentation import create_segmentation_message
 from .thermal import create_thermal_message
 from .tracked_features import create_tracked_features_message
-from .misc import create_age_gender_message
 
 __all__ = [
     "create_image_message",

depthai_nodes/ml/messages/creators/misc.py

@@ -1,28 +1,26 @@
-from typing import List, Union
-
-import depthai as dai
-import numpy as np
+from typing import List
 
 from ...messages import AgeGender
 
-def create_age_gender_message(
-    age: float,
-    gender_prob: List[float]
-) -> AgeGender:
-    """Create a message for the keypoints. The message contains 2D or 3D coordinates of
-    the detected keypoints.
 
-    Args:
-        age (float): Detected person age.
-        gender_prob (List[float]): Detected person gender probability [female, male].
+def create_age_gender_message(age: float, gender_prob: List[float]) -> AgeGender:
+    """Create a DepthAI message for the age and gender probability.
 
-    Returns:
-        AgeGender: Message containing the detected person age and gender probability.
+    @param age: Detected person age (must be multiplied by 100 to get years).
+    @type age: float
+    @param gender_prob: Detected person gender probability [female, male].
+    @type gender_prob: List[float]
+    @return: AgeGender message containing the detected person age and gender
+        probability.
+    @rtype: AgeGender
+    @raise ValueError: If age is not a float.
+    @raise ValueError: If gender_prob is not a list.
+    @raise ValueError: If each item in gender_prob is not a float.
     """
 
     if not isinstance(age, float):
         raise ValueError(f"age should be float, got {type(age)}.")
-    
+
     if not isinstance(gender_prob, List):
         raise ValueError(f"gender_prob should be list, got {type(gender_prob)}.")
     for item in gender_prob:
@@ -34,5 +32,5 @@ def create_age_gender_message(
     age_gender_message = AgeGender()
     age_gender_message.age = age
     age_gender_message.gender_prob = gender_prob
-    
-    return age_gender_message
+
+    return age_gender_message

depthai_nodes/ml/messages/misc.py

@@ -8,23 +8,23 @@ def __init__(self):
         super().__init__()
         self._age: float = None
         self._gender_prob: List[float] = None
-        
+
     @property
     def age(self) -> float:
         return self._age
-    
+
     @age.setter
     def age(self, value: float):
         if not isinstance(value, float):
             raise TypeError(
                 f"start_point must be of type float, instead got {type(value)}."
             )
         self._age = value
-    
+
     @property
     def gender_prob(self) -> List[float]:
         return self._gender_prob
-    
+
     @gender_prob.setter
     def gender_prob(self, value: List[float]):
         if not isinstance(value, List):
@@ -36,4 +36,4 @@ def gender_prob(self, value: List[float]):
                 raise TypeError(
                     f"gender_prob list values must be of type float, instead got {type(value)}."
                 )
-        self._gender_prob = value
+        self._gender_prob = value

depthai_nodes/ml/parsers/__init__.py

@@ -1,3 +1,5 @@
+from .age_gender import AgeGenderParser
+from .hrnet import HRNetParser
 from .image_output import ImageOutputParser
 from .keypoints import KeypointParser
 from .mediapipe_hand_landmarker import MPHandLandmarkParser
@@ -10,8 +12,6 @@
 from .thermal_image import ThermalImageParser
 from .xfeat import XFeatParser
 from .yunet import YuNetParser
-from .age_gender import AgeGenderParser
-from .hrnet import HRNetParser
 
 __all__ = [
     "ImageOutputParser",