Skip to content

Commit b11bc13

Browse files
committed
update to test image to account for new extra fields
1 parent 64a3bc9 commit b11bc13

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

tests/data/serialization/labelbox_v1/test_image.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@ def test_image(file_path):
2222

2323
# We are storing the media types now.
2424
payload['media_type'] = 'image'
25+
payload['Global Key'] = None
2526

2627
assert serialized.keys() == payload.keys()
2728

@@ -31,6 +32,8 @@ def test_image(file_path):
3132
elif key == 'Label':
3233
for annotation_a, annotation_b in zip(serialized[key]['objects'],
3334
payload[key]['objects']):
35+
annotation_b['page'] = None
36+
annotation_b['unit'] = None
3437
if not len(annotation_a['classifications']):
3538
# We don't add a classification key to the payload if there is no classifications.
3639
annotation_a.pop('classifications')

0 commit comments

Comments
 (0)