Skip to content

Commit bf624fd

Browse files
author
Kevin Kim
committed
Add assertion for label creation state
1 parent da66ff5 commit bf624fd

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/integration/conftest.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
from labelbox.orm import query
1616
from labelbox.pagination import PaginatedCollection
1717
from labelbox.schema.annotation_import import LabelImport
18+
from labelbox.schema.enums import AnnotationImportState
1819
from labelbox.schema.invite import Invite
1920
from labelbox.schema.queue_mode import QueueMode
2021
from labelbox.schema.user import User
@@ -334,6 +335,7 @@ def create_label():
334335
upload_task = LabelImport.create_from_objects(
335336
client, project.uid, f'label-import-{uuid.uuid4()}', predictions)
336337
upload_task.wait_until_done(sleep_time_seconds=5)
338+
assert(upload_task.state == AnnotationImportState.FAILED)
337339

338340
project.create_label = create_label
339341
project.create_label()

0 commit comments

Comments
 (0)