Skip to content

Commit 3bf53d7

Browse files
authored
[PLT-2290] Fix for mmc test global key (#1936)
1 parent a9d9360 commit 3bf53d7

File tree

1 file changed

+6
-5
lines changed

1 file changed

+6
-5
lines changed

libs/labelbox/tests/integration/test_mmc_data_rows.py

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -23,10 +23,11 @@ def mmc_data_row(dataset):
2323

2424

2525
@pytest.fixture
26-
def mmc_data_row_all(dataset, make_metadata_fields, embedding):
26+
def mmc_data_row_all(dataset, make_metadata_fields, embedding, rand_gen):
2727
data = ModelEvaluationTemplate()
2828
data.row_data.rootMessageIds = ["root1"]
29-
data.global_key = "global_key"
29+
global_key = f"global_key_{rand_gen(str)}"
30+
data.global_key = global_key
3031
vector = [random.uniform(1.0, 2.0) for _ in range(embedding.dims)]
3132
data.embeddings = [{"embedding_id": embedding.id, "vector": vector}]
3233
data.metadata_fields = make_metadata_fields
@@ -39,7 +40,7 @@ def mmc_data_row_all(dataset, make_metadata_fields, embedding):
3940

4041
data_row = list(dataset.data_rows())[0]
4142

42-
yield data_row
43+
yield data_row, global_key
4344

4445
data_row.delete()
4546

@@ -57,7 +58,7 @@ def test_mmc(mmc_data_row):
5758

5859

5960
def test_mmc_all(mmc_data_row_all, embedding, constants):
60-
data_row = mmc_data_row_all
61+
data_row, global_key = mmc_data_row_all
6162
assert json.loads(data_row.row_data) == {
6263
"type": "application/vnd.labelbox.conversational.model-chat-evaluation",
6364
"draft": True,
@@ -66,7 +67,7 @@ def test_mmc_all(mmc_data_row_all, embedding, constants):
6667
"messages": {},
6768
"version": 2,
6869
}
69-
assert data_row.global_key == "global_key"
70+
assert data_row.global_key == global_key
7071
metadata_fields = data_row.metadata_fields
7172
metadata = data_row.metadata
7273
assert len(metadata_fields) == 3

0 commit comments

Comments
 (0)