Merge pull request #79 from scaleapi/cuboid_error_messaging

Add error messaging for cuboids in Python client

Author: ardila

nucleus/__init__.py

@@ -68,6 +68,7 @@
     Segment,
     SegmentationAnnotation,
     Point,
+    CuboidAnnotation,
 )
 from .constants import (
     ANNOTATION_METADATA_SCHEMA_KEY,
@@ -115,6 +116,7 @@
 )
 from .prediction import (
     BoxPrediction,
+    CuboidPrediction,
     PolygonPrediction,
     SegmentationPrediction,
 )
@@ -565,6 +567,8 @@ def annotate_dataset(
         :param update: whether to update or ignore conflicting annotations
         :return: {"dataset_id: str, "annotations_processed": int}
         """
+        if any((isinstance(ann, CuboidAnnotation) for ann in annotations)):
+            raise NotImplementedError("Cuboid annotations not yet supported")
 
         # Split payload into segmentations and Box/Polygon
         segmentations = [
@@ -723,6 +727,9 @@ def predict(
             "predictions_ignored": int,
         }
         """
+        if any((isinstance(ann, CuboidPrediction) for ann in annotations)):
+            raise NotImplementedError("Cuboid predictions not yet supported")
+
         segmentations = [
             ann
             for ann in annotations

nucleus/dataset.py

@@ -9,7 +9,11 @@
     serialize_and_write_to_presigned_url,
 )
 
-from .annotation import Annotation, check_all_annotation_paths_remote
+from .annotation import (
+    Annotation,
+    CuboidAnnotation,
+    check_all_annotation_paths_remote,
+)
 from .constants import (
     DATASET_ITEM_IDS_KEY,
     DATASET_LENGTH_KEY,
@@ -163,6 +167,9 @@ def annotate(
             "ignored_items": int,
         }
         """
+        if any((isinstance(ann, CuboidAnnotation) for ann in annotations)):
+            raise NotImplementedError("Cuboid annotations not yet supported")
+
         if asynchronous:
             check_all_annotation_paths_remote(annotations)