Specify labels attribute type in python

Author: sovrasov

src/python/model_api/models/anomaly.py

@@ -153,7 +153,7 @@ def parameters(cls) -> dict:
                     description="Task type",
                     default_value="segmentation",
                 ),
-                "labels": ListValue(description="List of class labels"),
+                "labels": ListValue(description="List of class labels", value_type=str),
             },
         )
         return parameters

src/python/model_api/models/classification.py

@@ -171,7 +171,7 @@ def parameters(cls) -> dict:
                     min=1,
                     description="Number of most likely labels",
                 ),
-                "labels": ListValue(description="List of class labels"),
+                "labels": ListValue(description="List of class labels", value_type=str),
                 "path_to_labels": StringValue(
                     description="Path to file with labels. Overrides the labels, if they sets via 'labels' parameter",
                 ),

src/python/model_api/models/detection_model.py

@@ -58,7 +58,7 @@ def parameters(cls):
                     default_value=0.5,
                     description="Probability threshold value for bounding box filtering",
                 ),
-                "labels": ListValue(description="List of class labels"),
+                "labels": ListValue(description="List of class labels", value_type=str),
                 "path_to_labels": StringValue(
                     description="Path to file with labels. Overrides the labels, if they sets via 'labels' parameter",
                 ),

src/python/model_api/models/instance_segmentation.py

@@ -40,7 +40,7 @@ def parameters(cls) -> dict:
                     default_value=0.5,
                     description="Probability threshold value for bounding box filtering",
                 ),
-                "labels": ListValue(description="List of class labels"),
+                "labels": ListValue(description="List of class labels", value_type=str),
                 "path_to_labels": StringValue(
                     description="Path to file with labels. Overrides the labels, if they sets via `labels` parameter",
                 ),

src/python/model_api/models/segmentation.py

@@ -117,7 +117,7 @@ def parameters(cls) -> dict:
         parameters = super().parameters()
         parameters.update(
             {
-                "labels": ListValue(description="List of class labels"),
+                "labels": ListValue(description="List of class labels", value_type=str),
                 "path_to_labels": StringValue(
                     description="Path to file with labels. Overrides the labels, if they sets via 'labels' parameter",
                 ),