fix dicom test

Author: Dmitriy Apollonin

Makefile

@@ -1,6 +1,6 @@
 
 build-image:
-	docker build -t local/labelbox-python:test .
+	docker build --quiet -t local/labelbox-python:test .
 
 test-local: build-image
 

tests/data/serialization/ndjson/test_dicom.py

@@ -1,5 +1,6 @@
 from copy import copy
 import pytest
+import base64
 import labelbox.types as lb_types
 from labelbox.data.serialization import NDJsonConverter
 from labelbox.data.serialization.ndjson.objects import NDDicomSegments, NDDicomSegment, NDDicomLine
@@ -89,9 +90,11 @@
     'masks': {
         'frames': [{
             'index': 1,
+            'imBytes': None,
             'instanceURI': instance_uri_1
         }, {
             'index': 5,
+            'imBytes': None,
             'instanceURI': instance_uri_5
         }],
         'instances': [
@@ -157,9 +160,12 @@
     video_mask_label_with_global_key
 ]
 ndjsons = [
-    polyline_annotation_ndjson, polyline_annotation_ndjson_with_global_key,
-    dicom_mask_annotation_ndjson, dicom_mask_annotation_ndjson_with_global_key,
-    video_mask_annotation_ndjson, video_mask_annotation_ndjson_with_global_key
+    polyline_annotation_ndjson,
+    polyline_annotation_ndjson_with_global_key,
+    dicom_mask_annotation_ndjson,
+    dicom_mask_annotation_ndjson_with_global_key,
+    video_mask_annotation_ndjson,
+    video_mask_annotation_ndjson_with_global_key,
 ]
 labels_ndjsons = list(zip(labels, ndjsons))
 

tests/integration/annotation_import/test_data_types.py

@@ -65,6 +65,7 @@ def get_annotation_comparison_dicts_from_labels(labels):
         if 'masks' in annotation:
             for frame in annotation['masks']['frames']:
                 frame.pop('instanceURI')
+                frame.pop('imBytes')
             for instance in annotation['masks']['instances']:
                 instance.pop('colorRGB')
     return labels_ndjson