stash

Author: jihan-yin

nucleus/annotation.py

@@ -34,6 +34,7 @@ def __init__(
         height: Union[float, int],
         reference_id: str = None,
         item_id: str = None,
+        annotation_id: str = None,
         metadata: Optional[Dict] = None,
     ):
         if bool(reference_id) == bool(item_id):
@@ -47,6 +48,7 @@ def __init__(
         self.height = height
         self.reference_id = reference_id
         self.item_id = item_id
+        self.annotation_id = annotation_id
         self.metadata = metadata if metadata else {}
 
     @classmethod
@@ -60,6 +62,7 @@ def from_json(cls, payload: dict):
             height=geometry.get(HEIGHT_KEY, 0),
             reference_id=payload.get(REFERENCE_ID_KEY, None),
             item_id=payload.get(DATASET_ITEM_ID_KEY, None),
+            annotation_id=payload.get(ANNOTATION_ID, None),
             metadata=payload.get(METADATA_KEY, {}),
         )
 
@@ -74,6 +77,7 @@ def to_payload(self) -> dict:
                 HEIGHT_KEY: self.height,
             },
             REFERENCE_ID_KEY: self.reference_id,
+            ANNOTATION_ID: self.annotation_id,
             METADATA_KEY: self.metadata,
         }
 
@@ -89,6 +93,7 @@ def __init__(
         vertices: List[Any],
         reference_id: str = None,
         item_id: str = None,
+        annotation_id: str = None,
         metadata: Optional[Dict] = None,
     ):
         if bool(reference_id) == bool(item_id):
@@ -99,6 +104,7 @@ def __init__(
         self.vertices = vertices
         self.reference_id = reference_id
         self.item_id = item_id
+        self.annotation_id = annotation_id
         self.metadata = metadata if metadata else {}
 
     @classmethod
@@ -109,6 +115,7 @@ def from_json(cls, payload: dict):
             vertices=geometry.get(VERTICES_KEY, []),
             reference_id=payload.get(REFERENCE_ID_KEY, None),
             item_id=payload.get(DATASET_ITEM_ID_KEY, None),
+            annotation_id=payload.get(ANNOTATION_ID, None),
             metadata=payload.get(METADATA_KEY, {}),
         )
 
@@ -118,6 +125,7 @@ def to_payload(self) -> dict:
             TYPE_KEY: POLYGON_TYPE,
             GEOMETRY_KEY: {VERTICES_KEY: self.vertices},
             REFERENCE_ID_KEY: self.reference_id,
+            ANNOTATION_ID: self.annotation_id,
             METADATA_KEY: self.metadata,
         }
 

nucleus/constants.py

@@ -1,4 +1,5 @@
-NUCLEUS_ENDPOINT = "https://api.scale.com/v1/nucleus"
+#NUCLEUS_ENDPOINT = "https://api.scale.com/v1/nucleus"
+NUCLEUS_ENDPOINT = "http://localhost:3000/v1/nucleus"
 ITEMS_KEY = "items"
 ITEM_KEY = "item"
 REFERENCE_ID_KEY = "reference_id"