Remove NDSegments

Author: Val Brodsky

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

@@ -38,7 +38,6 @@
 from .objects import (
     NDObject,
     NDObjectType,
-    NDSegments,
     NDVideoMasks,
 )
 from .relationship import NDRelationship
@@ -49,7 +48,6 @@
     NDPromptClassificationType,
     NDConfusionMatrixMetric,
     NDScalarMetric,
-    NDSegments,
     NDVideoMasks,
     NDRelationship,
     NDPromptText,
@@ -133,15 +131,7 @@ 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, NDSegments):
-                    annotations.extend(
-                        NDSegments.to_common(
-                            ndjson_annotation,
-                            ndjson_annotation.name,
-                            ndjson_annotation.schema_id,
-                        )
-                    )
-                elif isinstance(ndjson_annotation, NDVideoMasks):
+                if isinstance(ndjson_annotation, NDVideoMasks):
                     annotations.append(
                         NDVideoMasks.to_common(ndjson_annotation)
                     )

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

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