Merge pull request #727 from Labelbox/kkim/wait-for-label-export

[AL-0] Add sleep before label exporter

Author: kkim-labelbox

tests/integration/test_export.py

@@ -1,3 +1,4 @@
+from time import sleep
 import uuid
 
 from labelbox.data.annotation_types.annotation import ObjectAnnotation
@@ -46,6 +47,8 @@ def test_export_annotations_nested_checklist(
     task = LabelImport.create_from_objects(client, project.uid,
                                            f'label-import-{uuid.uuid4()}', data)
     task.wait_until_done()
+    # Wait for exporter to retrieve latest labels
+    sleep(10)
     labels = project.label_generator().as_list()
     object_annotation = [
         annot for annot in next(labels).annotations

tests/integration/test_label.py

@@ -31,6 +31,8 @@ def test_labels(configured_project_with_label):
 def test_label_export(configured_project_with_label):
     project, _, _, label = configured_project_with_label
     label_id = label.uid
+    # Wait for exporter to retrieve latest labels
+    time.sleep(10)
 
     exported_labels_url = project.export_labels()
     assert exported_labels_url is not None