diff --git a/libs/labelbox/src/labelbox/schema/data_row_payload_templates.py b/libs/labelbox/src/labelbox/schema/data_row_payload_templates.py index 2e2728daa..194fb3321 100644 --- a/libs/labelbox/src/labelbox/schema/data_row_payload_templates.py +++ b/libs/labelbox/src/labelbox/schema/data_row_payload_templates.py @@ -15,7 +15,6 @@ class ModelEvalutationTemplateRowData(BaseModel): actors: Dict = Field(default={}) version: int = Field(default=2, frozen=True) messages: Dict = Field(default={}) - global_key: Optional[str] = None class ModelEvaluationTemplate(BaseModel): @@ -36,6 +35,7 @@ class ModelEvaluationTemplate(BaseModel): row_data: ModelEvalutationTemplateRowData = Field( default=ModelEvalutationTemplateRowData() ) + global_key: Optional[str] = None attachments: List[Dict] = Field(default=[]) embeddings: List[Dict] = Field(default=[]) metadata_fields: List[DataRowMetadataField] = Field(default=[]) diff --git a/libs/labelbox/tests/integration/test_mmc_data_rows.py b/libs/labelbox/tests/integration/test_mmc_data_rows.py index 2fa7bdd1b..ed4f02dee 100644 --- a/libs/labelbox/tests/integration/test_mmc_data_rows.py +++ b/libs/labelbox/tests/integration/test_mmc_data_rows.py @@ -26,7 +26,7 @@ def mmc_data_row(dataset): def mmc_data_row_all(dataset, make_metadata_fields, embedding): data = ModelEvaluationTemplate() data.row_data.rootMessageIds = ["root1"] - data.row_data.global_key = "global_key" + data.global_key = "global_key" vector = [random.uniform(1.0, 2.0) for _ in range(embedding.dims)] data.embeddings = [{"embedding_id": embedding.id, "vector": vector}] data.metadata_fields = make_metadata_fields @@ -65,9 +65,8 @@ def test_mmc_all(mmc_data_row_all, embedding, constants): "actors": {}, "messages": {}, "version": 2, - "globalKey": "global_key", } - + assert data_row.global_key == "global_key" metadata_fields = data_row.metadata_fields metadata = data_row.metadata assert len(metadata_fields) == 3