File tree Expand file tree Collapse file tree 2 files changed +4
-6
lines changed Expand file tree Collapse file tree 2 files changed +4
-6
lines changed Original file line number Diff line number Diff line change 1
1
import pytest
2
2
3
3
from labelbox import Dataset , Project
4
- from labelbox .schema .project import QueueMode
5
4
6
5
IMAGE_URL = "https://storage.googleapis.com/diagnostics-demo-data/coco/COCO_train2014_000000000034.jpg"
7
6
@@ -21,7 +20,7 @@ def big_dataset(dataset: Dataset):
21
20
22
21
23
22
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 )
25
24
26
25
data_rows = [dr .uid for dr in list (big_dataset .export_data_rows ())]
27
26
batch = configured_project .create_batch ("test-batch" , data_rows , 3 )
Original file line number Diff line number Diff line change 5
5
6
6
from labelbox import Project , LabelingFrontend
7
7
from labelbox .exceptions import InvalidQueryError
8
- from labelbox .schema .project import QueueMode
9
8
10
9
11
10
def test_project (client , rand_gen ):
@@ -181,6 +180,6 @@ def test_queued_data_row_export(configured_project):
181
180
182
181
183
182
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
You can’t perform that action at this time.
0 commit comments