Fix tests

Author: Val Brodsky

libs/labelbox/src/labelbox/schema/internal/data_row_upsert_item.py

@@ -5,6 +5,7 @@
 from labelbox.pydantic_compat import BaseModel
 from labelbox.schema.identifiable import UniqueId, GlobalKey
 from labelbox import pydantic_compat
+from labelbox.schema.data_row import DataRow
 
 
 class DataRowItemBase(ABC, pydantic_compat.BaseModel):
@@ -78,7 +79,8 @@ def is_empty(self) -> bool:
             or the only key is `dataset_id`.
         :return: bool
         """
-        row_data = self.payload.get("row_data", None)
+        row_data = self.payload.get("row_data", None) or self.payload.get(
+            DataRow.row_data, None)
         return (not self.payload or len(self.payload.keys()) == 1 and
                 "dataset_id" in self.payload or row_data is None or
                 len(row_data) == 0)

libs/labelbox/tests/unit/test_data_row_upsert_data.py

@@ -7,6 +7,7 @@
 from labelbox.schema.asset_attachment import AttachmentType
 from labelbox.schema.dataset import Dataset
 from labelbox.schema.internal.descriptor_file_creator import DescriptorFileCreator
+from labelbox.schema.data_row import DataRow
 
 
 @pytest.fixture
@@ -119,6 +120,11 @@ def test_create_is_empty():
         id={}, payload={"row_data": "http://my_site.com/photos/img_01.jpg"})
     assert not item.is_empty()
 
+    item = DataRowCreateItem(
+        id={},
+        payload={DataRow.row_data: "http://my_site.com/photos/img_01.jpg"})
+    assert not item.is_empty()
+
 
 def test_create_row_data_none():
     items = [