fix tests

Author: Matt Sokoloff

labelbox/schema/data_row.py

@@ -61,12 +61,15 @@ class DataRow(DbObject, Updateable, BulkDeletable):
     supported_meta_types = supported_attachment_types = set(
         Entity.AssetAttachment.AttachmentType.__members__)
 
-    def __init__(self, client, field_values, **kwargs):
-        field_values.update({'mediaType': MediaType.Unknown})
-        super().__init__(client, field_values, **kwargs)
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
         self.attachments.supports_filtering = False
         self.attachments.supports_sorting = False
 
+    def _set_field_values(self, field_values):
+        field_values.update({'mediaType': MediaType.Unknown})
+        super()._set_field_values(field_values)
+
     @staticmethod
     def bulk_delete(data_rows) -> None:
         """ Deletes all the given DataRows.

labelbox/schema/dataset.py

@@ -353,8 +353,8 @@ def validate_keys(item):
 
         def formatLegacyConversationalData(item):
             messages = item.pop("conversationalData")
-            version = item.pop("version")
-            type = item.pop("type")
+            version = item.pop("version", 1)
+            type = item.pop("type", "application/vnd.labelbox.conversational")
             if "externalId" in item:
                 external_id = item.pop("externalId")
                 item["external_id"] = external_id

labelbox/test.py

@@ -0,0 +1,18 @@
+def run(fn, model_run):
+    try:
+        fn()
+    except Exception as e:
+        model_run.update_status(error_message=error)
+        pipelines[pipeline].update_status(PipelineState.FAILED,
+                                          json_data['model_run_id'],
+                                          error_message=str(e))
+    else:
+        status
+
+
+def model_run(payload):
+
+    def etl():
+        payload
+
+    run(etl)

tests/integration/test_data_row_media_attributes.py

@@ -7,4 +7,4 @@ def test_export_empty_media_attributes(configured_project_with_label):
     sleep(10)
     labels = project.label_generator()
     label = next(labels)
-    assert label.data.media_attributes == {}
+    assert label.data.media_attributes == {}

tests/integration/test_data_rows.py

@@ -719,10 +719,12 @@ def test_create_conversational_text(dataset):
     }
     examples = [
         {
-            **content, 'media_type': 'CONVERSATIONAL_TEXT'
+            **content, 'media_type': 'CONVERSATIONAL'
         },
         content,
-        content['row_data']  # Old way to check for backwards compatibility
+        {
+            "conversationalData": content['row_data']['messages']
+        }  # Old way to check for backwards compatibility
     ]
     dataset.create_data_rows_sync(examples)
     data_rows = list(dataset.data_rows())
@@ -754,7 +756,7 @@ def test_invalid_media_type(dataset):
             "Found invalid contents for media type: 'IMAGE'", 'IMAGE'
     ], ["Found invalid media type: 'totallyinvalid'", 'totallyinvalid']]:
         # TODO: What error kind should this be? It looks like for global key we are
-        # using malformed query. But for FileUploads we use InvalidQueryError
+        # using malformed query. But for invalid contents in FileUploads we use InvalidQueryError
         with pytest.raises(labelbox.exceptions.InvalidQueryError):
             dataset.create_data_rows_sync([{
                 **content, 'media_type': invalid_media_type