format and add cuid type

Author: Matt Sokoloff

labelbox/data/annotation_types/annotation.py

@@ -49,9 +49,5 @@ class VideoClassificationAnnotation(ClassificationAnnotation):
     frame: int
 
 
-AnnotationType = Union[ClassificationAnnotation, ObjectAnnotation]
-VideoAnnotationType = Union[VideoObjectAnnotation,
-                            VideoClassificationAnnotation]
-
 VideoObjectAnnotation.update_forward_refs()
 ObjectAnnotation.update_forward_refs()

labelbox/data/annotation_types/feature.py

@@ -2,6 +2,8 @@
 
 from pydantic import BaseModel, root_validator
 
+from .types import Cuid
+
 
 class FeatureSchema(BaseModel):
     """
@@ -13,7 +15,7 @@ class FeatureSchema(BaseModel):
     to retroactively add schema ids by looking them up from the names.
     """
     name: Optional[str] = None
-    schema_id: Optional[str] = None
+    schema_id: Optional[Cuid] = None
 
     @root_validator
     def must_set_one(cls, values):

labelbox/data/annotation_types/label.py

@@ -8,13 +8,15 @@
 from .data import VideoData, TextData, RasterData
 from .geometry.mask import Mask
 from .metrics import Metric
-from .annotation import (AnnotationType, ClassificationAnnotation,
-                         ObjectAnnotation, VideoAnnotationType)
+from .annotation import (ClassificationAnnotation, ObjectAnnotation,
+                         VideoClassificationAnnotation, VideoObjectAnnotation)
 
 
 class Label(BaseModel):
     data: Union[VideoData, RasterData, TextData]
-    annotations: List[Union[AnnotationType, VideoAnnotationType, Metric]] = []
+    annotations: List[Union[ClassificationAnnotation, ObjectAnnotation,
+                            VideoObjectAnnotation,
+                            VideoClassificationAnnotation, Metric]] = []
     extra: Dict[str, Any] = {}
 
     def add_url_to_data(self, signer) -> "Label":

labelbox/data/annotation_types/types.py

@@ -0,0 +1,4 @@
+from pydantic import Field
+from typing_extensions import Annotated
+
+Cuid = Annotated[str, Field(min_length=25, max_length=25)]