Skip to content

Commit b13d4e5

Browse files
author
Kevin Kim
committed
Fix test failure due to QueueMode
1 parent 065d2d7 commit b13d4e5

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

tests/integration/annotation_import/conftest.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,8 @@
77

88
from typing import Type
99
from labelbox.schema.labeling_frontend import LabelingFrontend
10-
from labelbox.schema.annotation_import import LabelImport, MALPredictionImport, AnnotationImportState
10+
from labelbox.schema.annotation_import import LabelImport, AnnotationImportState
11+
from labelbox.schema.queue_mode import QueueMode
1112

1213

1314
@pytest.fixture
@@ -164,7 +165,7 @@ def configured_project_without_data_rows(client, configured_project, rand_gen):
164165
client.get_labeling_frontends(
165166
where=LabelingFrontend.name == "editor"))[0]
166167
project.setup_editor(configured_project.ontology())
167-
project.update(queue_mode=project.QueueMode.Batch)
168+
project.update(queue_mode=QueueMode.Batch)
168169
yield project
169170
project.delete()
170171

0 commit comments

Comments
 (0)