Skip to content

Commit 85bf5e3

Browse files
author
Kevin Kim
committed
Revert project.create_label() since some tests depend on it
1 parent fe955a2 commit 85bf5e3

File tree

1 file changed

+11
-5
lines changed

1 file changed

+11
-5
lines changed

tests/integration/conftest.py

Lines changed: 11 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -337,11 +337,17 @@ def configured_project_with_label(client, rand_gen, image_url, project, dataset,
337337
}
338338
}]
339339

340-
upload_task = LabelImport.create_from_objects(
341-
client, project.uid, f'label-import-{uuid.uuid4()}', predictions)
342-
upload_task.wait_until_done(sleep_time_seconds=5)
343-
assert upload_task.state == AnnotationImportState.FINISHED, "Label Import failed"
344-
340+
def create_label():
341+
""" Ad-hoc function to create a LabelImport
342+
Creates a LabelImport task which will create a label
343+
"""
344+
upload_task = LabelImport.create_from_objects(
345+
client, project.uid, f'label-import-{uuid.uuid4()}', predictions)
346+
upload_task.wait_until_done(sleep_time_seconds=5)
347+
assert upload_task.state == AnnotationImportState.FINISHED, "Label Import failed"
348+
349+
project.create_label = create_label
350+
project.create_label()
345351
label = project.labels().get_one()
346352
assert label is not None, "Cannot fetch created label"
347353
yield [project, dataset, datarow, label]

0 commit comments

Comments
 (0)