Fix - NN Archive Validation (#123)

Fixed validation errors in NN Archive.

Author: kozlov721

luxonis_ml/nn_archive/config_building_blocks/base_models/head.py

@@ -169,7 +169,7 @@ class HeadYOLO(HeadObjectDetection, HeadSegmentation, ABC):
         description="A configuration specifying which output names from the `outputs` block of the archive are fed into the head."
     )
     subtype: ObjectDetectionSubtypeYOLO = Field(
-        description="YOLO family decoding subtype (e.g. v5, v6, v7 etc.)."
+        description="YOLO family decoding subtype (e.g. yolov5, yolov6, yolov7 etc.)."
     )
     postprocessor_path: Optional[str] = Field(
         None,
@@ -276,7 +276,13 @@ def validate_task_specific_fields(
 
         # Validate Outputs
         outputs = values.get("outputs", {})
-        defined_params = {k for k, v in outputs.dict().items() if v is not None}
+        defined_params = {
+            k
+            for k, v in (
+                outputs.model_dump() if isinstance(outputs, BaseModel) else outputs
+            ).items()
+            if v is not None
+        }
 
         supported_output_params = {
             "instance_segmentation": ["yolo_outputs", "mask_outputs", "protos"],

luxonis_ml/nn_archive/config_building_blocks/base_models/head_outputs.py

@@ -51,15 +51,15 @@ class OutputsYOLO(Outputs):
 
     # Instance segmentation
     mask_outputs: Optional[List[str]] = Field(
-        description="A list of output names for each mask output."
+        None, description="A list of output names for each mask output."
     )
-    protos: Optional[str] = Field(description="Output name for the protos.")
+    protos: Optional[str] = Field(None, description="Output name for the protos.")
 
     # Keypoint detection
-    keypoints: Optional[str] = Field(description="Output name for the keypoints.")
+    keypoints: Optional[str] = Field(None, description="Output name for the keypoints.")
 
     # OBB detection
-    angles: Optional[str] = Field(description="Output name for the angles.")
+    angles: Optional[str] = Field(None, description="Output name for the angles.")
 
 
 class OutputsSSD(Outputs):

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "luxonis-ml"
-version = "0.2.0"
+version = "0.2.1"
 description = "MLOps tools for training models for Luxonis devices"
 readme = "README.md"
 requires-python = ">=3.8"