fixes remap labels from raising an error to just not using them

NKI-AI · Nov 24, 2023 · 356f4a2 · 356f4a2
1 parent 58a200c
commit 356f4a2
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/ahcore/callbacks.py b/ahcore/callbacks.py
@@ -195,14 +195,13 @@ def _get_annotation_data(
         if not self._data_description:
             raise ValueError("No data description is provided.")
 
-        if not self._data_description.remap_labels:
-            raise ValueError("Remap labels are not provided.")
-
         if not self._data_description.index_map:
             raise ValueError("Index map is not provided.")
 
         _annotations = self._annotations.read_region(coordinates, self._scaling, self._region_size)
-        _annotations = rename_labels(_annotations, remap_labels=self._data_description.remap_labels)
+
+        if self._data_description.remap_labels:
+            _annotations = rename_labels(_annotations, remap_labels=self._data_description.remap_labels)
 
         points, boxes, region, roi = convert_annotations(
             _annotations,