gt box working

Author: jihan-yin

nucleus/constants.py

@@ -14,7 +14,6 @@
 ERROR_PAYLOAD = "error_payload"
 ERROR_CODES = "error_codes"
 ANNOTATIONS_KEY = "annotations"
-ANNOTATION_ITEMS_KEY = "annotation_items"
 ANNOTATION_ID_KEY = "annotation_id"
 ANNOTATIONS_PROCESSED_KEY = "annotations_processed"
 PREDICTIONS_PROCESSED_KEY = "predictions_processed"

nucleus/dataset.py

@@ -90,7 +90,7 @@ def create_model_run(
 
     def annotate(
         self,
-        annotations: List[Union[BoxAnnotation | PolygonAnnotation]],
+        annotations: List[Union[BoxAnnotation, PolygonAnnotation]],
         update: Optional[bool] = DEFAULT_ANNOTATION_UPDATE_MODE,
         batch_size: int = 20,
     ) -> dict:

nucleus/payload_constructor.py

@@ -8,7 +8,6 @@
     METADATA_KEY,
     REFERENCE_ID_KEY,
     ANNOTATIONS_KEY,
-    ANNOTATION_ITEMS_KEY,
     ITEMS_KEY,
     FORCE_KEY,
     MODEL_ID_KEY,
@@ -37,7 +36,7 @@ def construct_annotation_payload(
     for annotation_item in annotation_items:
         annotations.append(annotation_item.to_payload())
 
-    return {ANNOTATION_ITEMS_KEY: annotations, ANNOTATION_UPDATE_KEY: update}
+    return {ANNOTATIONS_KEY: annotations, ANNOTATION_UPDATE_KEY: update}
 
 
 def construct_box_predictions_payload(