CLI: Let function create-native submit label types to the server

cvat-ai · Jan 31, 2025 · be87003 · be87003
1 parent 1a90c5f
commit be87003
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/cvat-cli/src/cvat_cli/_internal/commands_functions.py b/cvat-cli/src/cvat_cli/_internal/commands_functions.py
@@ -62,6 +62,11 @@ def execute(
                         "name": label_spec.name,
                     }
                 )
+
+                if getattr(label_spec, "type", "any") != "any":
+                    # Add the type conditionally, to stay compatible with older
+                    # CVAT versions when the function doesn't define label types.
+                    remote_function["labels_v2"][-1]["type"] = label_spec.type
         else:
             raise cvataa.BadFunctionError(
                 f"Unsupported function spec type: {type(function.spec).__name__}"