cleanup of code

Author: jtsodapop

labelbox/data/serialization/ndjson/label.py

@@ -29,8 +29,6 @@ def to_common(self) -> LabelGenerator:
             grouped_annotations[annotation.data_row.id].append(annotation)
         return LabelGenerator(
             data=self._generate_annotations(grouped_annotations))
-        # c = LabelGenerator(data=self._generate_annotations(grouped_annotations))
-        # print(next(c), "Wefwefweefw\n")
 
     @classmethod
     def from_common(cls,
@@ -50,8 +48,6 @@ def _generate_annotations(
             annots = []
             for annotation in annotations:
                 if isinstance(annotation, NDSegments):
-                    # b = NDSegments.to_common(annotation, annotation.schema_id)
-                    # print(type(b), b)
                     annots.extend(
                         NDSegments.to_common(annotation, annotation.schema_id))
 
@@ -65,7 +61,6 @@ def _generate_annotations(
                 else:
                     raise TypeError(
                         f"Unsupported annotation. {type(annotation)}")
-            # print(type(annots[0]), annots[0])
             data = self._infer_media_type(annotations)(uid=data_row_id)
             yield Label(annotations=annots, data=data)
 
@@ -101,7 +96,6 @@ def _create_video_annotations(
                 (VideoClassificationAnnotation, VideoObjectAnnotation)):
                 video_annotations[annot.feature_schema_id].append(annot)
 
-        #break this into two groups, classifications, and then objects
         for annotation_group in video_annotations.values():
             consecutive_frames = cls._get_consecutive_frames(
                 sorted([annotation.frame for annotation in annotation_group]))

labelbox/data/serialization/ndjson/objects.py

@@ -3,7 +3,6 @@
 from typing import Any, Dict, List, Tuple, Union
 import base64
 import numpy as np
-from py import process
 
 from pydantic import BaseModel
 from PIL import Image
@@ -158,7 +157,6 @@ def lookup_segment_object_type(segment: List) -> "NDFrameObjectType":
         return result
 
     def to_common(self, feature_schema_id: Cuid):
-        # print(f"\n to_common in ndsegment {self.keyframes}\n")
         return [
             keyframe.to_common(feature_schema_id) for keyframe in self.keyframes
         ]