Skip to content

Commit 76bdadc

Browse files
author
Kevin Kim
committed
Reverting batch test changes
1 parent 12aa306 commit 76bdadc

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

tests/integration/test_batch.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,4 @@
11
import pytest
2-
import uuid
32

43
from labelbox import Dataset, Project
54

@@ -36,9 +35,8 @@ def test_create_batch(configured_project: Project, big_dataset: Dataset):
3635
configured_project.update(queue_mode=Project.QueueMode.Batch)
3736

3837
data_rows = [dr.uid for dr in list(big_dataset.export_data_rows())]
39-
batch_name = f"test-batch-{uuid.uuid4()}"
40-
batch = configured_project.create_batch(batch_name, data_rows, 3)
41-
assert batch.name == batch_name
38+
batch = configured_project.create_batch("test-batch", data_rows, 3)
39+
assert batch.name == "test-batch"
4240
assert batch.size == len(data_rows)
4341

4442

@@ -81,4 +79,4 @@ def test_export_data_rows(configured_project: Project, dataset: Dataset):
8179
exported_data_rows = [dr.uid for dr in result]
8280

8381
assert len(result) == n_data_rows
84-
assert set(data_rows) == set(exported_data_rows)
82+
assert set(data_rows) == set(exported_data_rows)

0 commit comments

Comments
 (0)