Skip to content

Commit 9b98910

Browse files
author
gdj0nes
committed
FIX: tests from nesting
1 parent 9097270 commit 9b98910

File tree

2 files changed

+4
-6
lines changed

2 files changed

+4
-6
lines changed

tests/integration/test_batch.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
import pytest
22

33
from labelbox import Dataset, Project
4-
from labelbox.schema.project import QueueMode
54

65
IMAGE_URL = "https://storage.googleapis.com/diagnostics-demo-data/coco/COCO_train2014_000000000034.jpg"
76

@@ -21,7 +20,7 @@ def big_dataset(dataset: Dataset):
2120

2221

2322
def test_create_batch(configured_project: Project, big_dataset: Dataset):
24-
configured_project.update(queue_mode=QueueMode.Batch)
23+
configured_project.update(queue_mode=Project.QueueMode.Batch)
2524

2625
data_rows = [dr.uid for dr in list(big_dataset.export_data_rows())]
2726
batch = configured_project.create_batch("test-batch", data_rows, 3)

tests/integration/test_project.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,6 @@
55

66
from labelbox import Project, LabelingFrontend
77
from labelbox.exceptions import InvalidQueryError
8-
from labelbox.schema.project import QueueMode
98

109

1110
def test_project(client, rand_gen):
@@ -181,6 +180,6 @@ def test_queued_data_row_export(configured_project):
181180

182181

183182
def test_queue_mode(configured_project: Project):
184-
assert configured_project.queue_mode() == QueueMode.Dataset
185-
configured_project.update(queue_mode=QueueMode.Batch)
186-
assert configured_project.queue_mode() == QueueMode.Batch
183+
assert configured_project.queue_mode() == configured_project.QueueMode.Dataset
184+
configured_project.update(queue_mode=configured_project.QueueMode.Batch)
185+
assert configured_project.queue_mode() == configured_project.QueueMode.Batch

0 commit comments

Comments
 (0)