Skip to content

Commit 12aa306

Browse files
author
Kevin Kim
committed
Randomize batch name for retestability
1 parent e10c156 commit 12aa306

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

tests/integration/test_batch.py

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

34
from labelbox import Dataset, Project
45

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

3738
data_rows = [dr.uid for dr in list(big_dataset.export_data_rows())]
38-
batch = configured_project.create_batch("test-batch", data_rows, 3)
39-
assert batch.name == 'test-batch'
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
4042
assert batch.size == len(data_rows)
4143

4244

0 commit comments

Comments
 (0)