Added: handling of no_label_annotation_set (#529)

konfuzio-ai · Aug 8, 2024 · e657f44 · e657f44
1 parent 8950d86
commit e657f44
Showing 1 changed file with 11 additions and 8 deletions.
diff --git a/konfuzio_sdk/data.py b/konfuzio_sdk/data.py
@@ -3137,14 +3137,17 @@ def no_label_annotation_set(self) -> AnnotationSet:
             self.annotation_sets()
 
             # Find no label annotation set
-            self._no_label_annotation_set = next(
-                (
-                    annotation_set
-                    for annotation_set in self._annotation_sets
-                    if annotation_set.label_set == self.project.no_label_set
-                ),
-                None,
-            )
+            if self._annotation_sets is not None:
+                self._no_label_annotation_set = next(
+                    (
+                        annotation_set
+                        for annotation_set in self._annotation_sets
+                        if annotation_set.label_set == self.project.no_label_set
+                    ),
+                    None,
+                )
+            else:
+                self._no_label_annotation_set = None
 
             # Create no label annotation set if not found
             if self._no_label_annotation_set is None: