Revert "Remove NDSegments"

This reverts commit a25b066.

Author: Val Brodsky

libs/labelbox/src/labelbox/data/serialization/ndjson/label.py

@@ -38,6 +38,7 @@
 from .objects import (
     NDObject,
     NDObjectType,
+    NDSegments,
     NDVideoMasks,
 )
 from .relationship import NDRelationship
@@ -48,6 +49,7 @@
     NDPromptClassificationType,
     NDConfusionMatrixMetric,
     NDScalarMetric,
+    NDSegments,
     NDVideoMasks,
     NDRelationship,
     NDPromptText,
@@ -131,7 +133,15 @@ def _generate_annotations(
             # deserialized objects in the _AnnotationGroupTuple
             # object *if* the object can be used in a relationship
             for uuid, ndjson_annotation in group.ndjson_annotations.items():
-                if isinstance(ndjson_annotation, NDVideoMasks):
+                if isinstance(ndjson_annotation, NDSegments):
+                    annotations.extend(
+                        NDSegments.to_common(
+                            ndjson_annotation,
+                            ndjson_annotation.name,
+                            ndjson_annotation.schema_id,
+                        )
+                    )
+                elif isinstance(ndjson_annotation, NDVideoMasks):
                     annotations.append(
                         NDVideoMasks.to_common(ndjson_annotation)
                     )

libs/labelbox/src/labelbox/data/serialization/ndjson/objects.py

@@ -724,6 +724,8 @@ def lookup_object(
     ) -> "NDObjectType":
         if isinstance(annotation, VideoMaskAnnotation):
             result = NDVideoMasks
+        elif isinstance(annotation, list):
+            result = NDSegments
         else:
             result = {
                 Line: NDLine,