Fixed default mapping when lambda function called (#8250)

cvat-ai · Aug 1, 2024 · f3dcc3e · f3dcc3e
1 parent e0da3a2
commit f3dcc3e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/changelog.d/20240801_104929_boris_fixed_default_mapping.md b/changelog.d/20240801_104929_boris_fixed_default_mapping.md
@@ -0,0 +1,4 @@
+### Fixed
+
+- API call to run automatic annotations fails on a model with attributes
+ when mapping not provided in the request (<https://github.com/cvat-ai/cvat/pull/8250>)
diff --git a/cvat/apps/lambda_manager/views.py b/cvat/apps/lambda_manager/views.py
@@ -271,9 +271,9 @@ def make_default_mapping(model_labels, task_labels):
                     if task_label.name == model_label['name'] and labels_compatible(model_label, task_label):
                         attributes_default_mapping = {}
                         for model_attr in model_label.get('attributes', {}):
-                            for db_attr in model_label.attributespec_set.all():
+                            for db_attr in task_label.attributespec_set.all():
                                 if db_attr.name == model_attr['name']:
-                                    attributes_default_mapping[model_attr] = db_attr.name
+                                    attributes_default_mapping[model_attr['name']] = db_attr.name
 
                         mapping_by_default[model_label['name']] = {
                             'name': task_label.name,