@@ -23,10 +23,11 @@ def mmc_data_row(dataset):
23
23
24
24
25
25
@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 ):
27
27
data = ModelEvaluationTemplate ()
28
28
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
30
31
vector = [random .uniform (1.0 , 2.0 ) for _ in range (embedding .dims )]
31
32
data .embeddings = [{"embedding_id" : embedding .id , "vector" : vector }]
32
33
data .metadata_fields = make_metadata_fields
@@ -39,7 +40,7 @@ def mmc_data_row_all(dataset, make_metadata_fields, embedding):
39
40
40
41
data_row = list (dataset .data_rows ())[0 ]
41
42
42
- yield data_row
43
+ yield data_row , global_key
43
44
44
45
data_row .delete ()
45
46
@@ -57,7 +58,7 @@ def test_mmc(mmc_data_row):
57
58
58
59
59
60
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
61
62
assert json .loads (data_row .row_data ) == {
62
63
"type" : "application/vnd.labelbox.conversational.model-chat-evaluation" ,
63
64
"draft" : True ,
@@ -66,7 +67,7 @@ def test_mmc_all(mmc_data_row_all, embedding, constants):
66
67
"messages" : {},
67
68
"version" : 2 ,
68
69
}
69
- assert data_row .global_key == " global_key"
70
+ assert data_row .global_key == global_key
70
71
metadata_fields = data_row .metadata_fields
71
72
metadata = data_row .metadata
72
73
assert len (metadata_fields ) == 3
0 commit comments