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

cvat-ai · Feb 7, 2025 · 567a46f · 567a46f
1 parent 1a90c5f
commit 567a46f
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/changelog.d/20250207_153348_roman_native_function_label_type.md b/changelog.d/20250207_153348_roman_native_function_label_type.md
@@ -0,0 +1,5 @@
+### Added
+
+- \[CLI\] `function create-native` now sends the function's declared label types
+  to the server
+  (<https://github.com/cvat-ai/cvat/pull/9035>)
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__}"