Fixed refloc bugs

Author: Claire Pajot

nucleus/annotation.py

@@ -43,6 +43,8 @@ def from_json(cls, payload: dict):
             return PolygonAnnotation.from_json(payload)
         elif payload.get(TYPE_KEY, None) == CUBOID_TYPE:
             return CuboidAnnotation.from_json(payload)
+        elif payload.get(TYPE_KEY, None) == CATEGORY_TYPE:
+            return CategoryAnnotation.from_json(payload)
         else:
             return SegmentationAnnotation.from_json(payload)
 

nucleus/constants.py

@@ -10,7 +10,13 @@
 CUBOID_TYPE = "cuboid"
 CATEGORY_TYPE = "category"
 MULTICATEGORY_TYPE = "multicategory"
-ANNOTATION_TYPES = (BOX_TYPE, POLYGON_TYPE, SEGMENTATION_TYPE, CUBOID_TYPE)
+ANNOTATION_TYPES = (
+    BOX_TYPE,
+    POLYGON_TYPE,
+    SEGMENTATION_TYPE,
+    CUBOID_TYPE,
+    CATEGORY_TYPE,
+)
 ANNOTATION_UPDATE_KEY = "update"
 AUTOTAGS_KEY = "autotags"
 EXPORTED_ROWS = "exportedRows"