diff --git a/src/superannotate/lib/core/usecases/images.py b/src/superannotate/lib/core/usecases/images.py index d7d978e6e..10abf0f09 100644 --- a/src/superannotate/lib/core/usecases/images.py +++ b/src/superannotate/lib/core/usecases/images.py @@ -1593,15 +1593,14 @@ def fill_classes_data(self, annotations: dict): attribute["groupName"] = annotation_class["attribute_groups"][ attribute["groupId"] ]["name"] - if attribute["id"] not in list( + if attribute.get("id") in list( annotation_class["attribute_groups"][attribute["groupId"]][ "attributes" ].keys() ): - continue - attribute["name"] = annotation_class["attribute_groups"][ - attribute["groupId"] - ]["attributes"][attribute["id"]] + attribute["name"] = annotation_class["attribute_groups"][ + attribute["groupId"] + ]["attributes"][attribute["id"]] def execute(self): if self.is_valid():