[PLT-383] Fix SDK integration tests.

Author: sfendell-labelbox

tests/integration/annotation_import/fixtures/export_v2.py

@@ -144,8 +144,12 @@ def expected_export_v2_text():
             'annotation_kind': 'TextEntity',
             'classifications': [],
             'location': {
-                'start': 67,
-                'end': 128
+                'start':
+                    67,
+                'end':
+                    128,
+                'token':
+                    'ore people to express themselves online😞😂, research suggests',
             }
         }],
         'classifications': [{
@@ -174,9 +178,7 @@ def expected_export_v2_text():
 def expected_export_v2_video():
     expected_annotations = {
         'frames': {},
-        'segments': {
-            '<cuid>': [[7, 13], [18, 19]]
-        },
+        'segments': {},
         'key_frame_feature_map': {},
         'classifications': [{
             'name':

tests/integration/annotation_import/test_data_types.py

@@ -296,7 +296,9 @@ def test_import_data_types_v2(client, configured_project, initial_dataset,
     # to be similar to tests/integration/test_task_queue.py
 
     result = export_v2_test_helpers.run_project_export_v2_task(project)
-    exported_data = result[0]
+    find_data_row = lambda dr: dr['data_row']['id'] == data_row.uid
+    exported_data = list(filter(find_data_row, result))[0]
+    assert exported_data
 
     # timestamp fields are in iso format
     validate_iso_format(exported_data['data_row']['details']['created_at'])