From 33c4696efd760a95c448aa61a852d2595ba94483 Mon Sep 17 00:00:00 2001 From: Oleg Trygub Date: Fri, 2 Sep 2022 21:39:48 -0400 Subject: [PATCH 1/2] Added globalkey support to bulk import --- labelbox/schema/dataset.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/labelbox/schema/dataset.py b/labelbox/schema/dataset.py index 954dba8c3..a2deb62ab 100644 --- a/labelbox/schema/dataset.py +++ b/labelbox/schema/dataset.py @@ -360,6 +360,9 @@ def convert_item(item): if "externalId" in item: external_id = item.pop("externalId") item["external_id"] = external_id + if "globalKey" in item: + global_key = item.pop("globalKey") + item["globalKey"] = global_key validate_conversational_data(messages) one_conversation = \ { From 6ae6ac8e1ecc8de2c72bb22091b6dfcf1a4a2780 Mon Sep 17 00:00:00 2001 From: jtsodapop <67922677+jtsodapop@users.noreply.github.com> Date: Tue, 6 Sep 2022 12:51:51 -0400 Subject: [PATCH 2/2] add test cases for creation of data rows. pending api side updates to affirm these are working --- tests/integration/test_data_rows.py | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) diff --git a/tests/integration/test_data_rows.py b/tests/integration/test_data_rows.py index cb64cd9be..0049434c2 100644 --- a/tests/integration/test_data_rows.py +++ b/tests/integration/test_data_rows.py @@ -582,3 +582,25 @@ def test_create_data_rows_local_file(dataset, sample_image): data_row = list(dataset.data_rows())[0] assert data_row.external_id == "tests/integration/media/sample_image.jpg" assert len(data_row.metadata_fields) == 4 + + +def test_create_data_row_with_global_key(dataset, sample_image): + global_key = str(uuid.uuid4()) + row = dataset.create_data_row(row_data=sample_image, global_key=global_key) + assert row.global_key == global_key + + +def test_create_data_rows_with_global_key(dataset, sample_image): + global_key_1 = str(uuid.uuid4()) + global_key_2 = str(uuid.uuid4()) + task = dataset.create_data_rows([{ + DataRow.row_data: sample_image, + DataRow.global_key: global_key_1 + }, { + DataRow.row_data: sample_image, + DataRow.global_key: global_key_2 + }]) + task.wait_till_done() + assert task.status == "COMPLETE" + created_global_keys = set([row.global_key for row in dataset.data_rows()]) + assert set([global_key_1, global_key_2]) == created_global_keys