hot fix added (#144)

luxonis · Dec 3, 2024 · da4fdb8 · da4fdb8
1 parent 71611c2
commit da4fdb8
Show file tree

Hide file tree

Showing 8 changed files with 47 additions and 32 deletions.
diff --git a/depthai_nodes/ml/messages/classification.py b/depthai_nodes/ml/messages/classification.py
@@ -112,8 +112,10 @@ def transformation(self, value: dai.ImgTransformation):
         @type value: dai.ImgTransformation
         @raise TypeError: If value is not a dai.ImgTransformation object.
         """
-        if not isinstance(value, dai.ImgTransformation):
-            raise TypeError(
-                f"Transformation must be a dai.ImgTransformation object, instead got {type(value)}."
-            )
+
+        if value is not None:
+            if not isinstance(value, dai.ImgTransformation):
+                raise TypeError(
+                    f"Transformation must be a dai.ImgTransformation object, instead got {type(value)}."
+                )
         self._transformation = value
diff --git a/depthai_nodes/ml/messages/clusters.py b/depthai_nodes/ml/messages/clusters.py
@@ -124,8 +124,10 @@ def transformation(self, value: dai.ImgTransformation):
         @type value: dai.ImgTransformation
         @raise TypeError: If value is not a dai.ImgTransformation object.
         """
-        if not isinstance(value, dai.ImgTransformation):
-            raise TypeError(
-                f"Transformation must be a dai.ImgTransformation object, instead got {type(value)}."
-            )
+
+        if value is not None:
+            if not isinstance(value, dai.ImgTransformation):
+                raise TypeError(
+                    f"Transformation must be a dai.ImgTransformation object, instead got {type(value)}."
+                )
         self._transformation = value
diff --git a/depthai_nodes/ml/messages/img_detections.py b/depthai_nodes/ml/messages/img_detections.py
@@ -247,8 +247,9 @@ def transformation(self, value: dai.ImgTransformation):
         @type value: dai.ImgTransformation
         @raise TypeError: If value is not a dai.ImgTransformation object.
         """
-        if not isinstance(value, dai.ImgTransformation):
-            raise TypeError(
-                f"Transformation must be a dai.ImgTransformation object, instead got {type(value)}."
-            )
+        if value is not None:
+            if not isinstance(value, dai.ImgTransformation):
+                raise TypeError(
+                    f"Transformation must be a dai.ImgTransformation object, instead got {type(value)}."
+                )
         self._transformation = value
diff --git a/depthai_nodes/ml/messages/keypoints.py b/depthai_nodes/ml/messages/keypoints.py
@@ -178,8 +178,10 @@ def transformation(self, value: dai.ImgTransformation):
         @type value: dai.ImgTransformation
         @raise TypeError: If value is not a dai.ImgTransformation object.
         """
-        if not isinstance(value, dai.ImgTransformation):
-            raise TypeError(
-                f"Transformation must be a dai.ImgTransformation object, instead got {type(value)}."
-            )
+
+        if value is not None:
+            if not isinstance(value, dai.ImgTransformation):
+                raise TypeError(
+                    f"Transformation must be a dai.ImgTransformation object, instead got {type(value)}."
+                )
         self._transformation = value
diff --git a/depthai_nodes/ml/messages/lines.py b/depthai_nodes/ml/messages/lines.py
@@ -151,8 +151,10 @@ def transformation(self, value: dai.ImgTransformation):
         @type value: dai.ImgTransformation
         @raise TypeError: If value is not a dai.ImgTransformation object.
         """
-        if not isinstance(value, dai.ImgTransformation):
-            raise TypeError(
-                f"Transformation must be a dai.ImgTransformation object, instead got {type(value)}."
-            )
+
+        if value is not None:
+            if not isinstance(value, dai.ImgTransformation):
+                raise TypeError(
+                    f"Transformation must be a dai.ImgTransformation object, instead got {type(value)}."
+                )
         self._transformation = value
diff --git a/depthai_nodes/ml/messages/map.py b/depthai_nodes/ml/messages/map.py
@@ -92,8 +92,10 @@ def transformation(self, value: dai.ImgTransformation):
         @type value: dai.ImgTransformation
         @raise TypeError: If value is not a dai.ImgTransformation object.
         """
-        if not isinstance(value, dai.ImgTransformation):
-            raise TypeError(
-                f"Transformation must be a dai.ImgTransformation object, instead got {type(value)}."
-            )
+
+        if value is not None:
+            if not isinstance(value, dai.ImgTransformation):
+                raise TypeError(
+                    f"Transformation must be a dai.ImgTransformation object, instead got {type(value)}."
+                )
         self._transformation = value
diff --git a/depthai_nodes/ml/messages/prediction.py b/depthai_nodes/ml/messages/prediction.py
@@ -110,8 +110,10 @@ def transformation(self, value: dai.ImgTransformation):
         @type value: dai.ImgTransformation
         @raise TypeError: If value is not a dai.ImgTransformation object.
         """
-        if not isinstance(value, dai.ImgTransformation):
-            raise TypeError(
-                f"Transformation must be a dai.ImgTransformation object, instead got {type(value)}."
-            )
+
+        if value is not None:
+            if not isinstance(value, dai.ImgTransformation):
+                raise TypeError(
+                    f"Transformation must be a dai.ImgTransformation object, instead got {type(value)}."
+                )
         self._transformation = value
diff --git a/depthai_nodes/ml/messages/segmentation.py b/depthai_nodes/ml/messages/segmentation.py
@@ -69,8 +69,10 @@ def transformation(self, value: dai.ImgTransformation):
         @type value: dai.ImgTransformation
         @raise TypeError: If value is not a dai.ImgTransformation object.
         """
-        if not isinstance(value, dai.ImgTransformation):
-            raise TypeError(
-                f"Transformation must be a dai.ImgTransformation object, instead got {type(value)}."
-            )
+
+        if value is not None:
+            if not isinstance(value, dai.ImgTransformation):
+                raise TypeError(
+                    f"Transformation must be a dai.ImgTransformation object, instead got {type(value)}."
+                )
         self._transformation = value