create data row from objects

Author: Matt Sokoloff

labelbox/orm/model.py

@@ -104,8 +104,8 @@ def DateTime(*args):
         return Field(Field.Type.DateTime, *args)
 
     @staticmethod
-    def Enum(enum_cls: type, *args):
-        return Field(Field.EnumType(enum_cls), *args)
+    def Enum(enum_cls: type, *args, **kwargs):
+        return Field(Field.EnumType(enum_cls), *args, **kwargs)
 
     @staticmethod
     def Json(*args):

labelbox/schema/data_row.py

@@ -5,6 +5,7 @@
 from labelbox.orm.db_object import DbObject, Updateable, BulkDeletable
 from labelbox.orm.model import Entity, Field, Relationship
 from labelbox.schema.data_row_metadata import DataRowMetadataField  # type: ignore
+from labelbox.schema.media_type import MediaType
 
 if TYPE_CHECKING:
     from labelbox import AssetAttachment
@@ -48,6 +49,7 @@ class DataRow(DbObject, Updateable, BulkDeletable):
                           name="metadata",
                           graphql_name="customMetadata",
                           result_subquery="customMetadata { schemaId value }")
+    media_type = Field.Enum(MediaType, "media_type", result_subquery="")
 
     # Relationships
     dataset = Relationship.ToOne("Dataset")
@@ -59,8 +61,9 @@ class DataRow(DbObject, Updateable, BulkDeletable):
     supported_meta_types = supported_attachment_types = set(
         Entity.AssetAttachment.AttachmentType.__members__)
 
-    def __init__(self, *args, **kwargs):
-        super().__init__(*args, **kwargs)
+    def __init__(self, client, field_values, **kwargs):
+        field_values.update({'mediaType': MediaType.Unknown})
+        super().__init__(client, field_values, **kwargs)
         self.attachments.supports_filtering = False
         self.attachments.supports_sorting = False
 

labelbox/schema/dataset.py

@@ -95,9 +95,12 @@ def convert_field_keys(items):
             raise InvalidQueryError(
                 "DataRow.row_data missing when creating DataRow.")
 
-        # If row data is a local file path, upload it to server.
         row_data = args[DataRow.row_data.name]
-        if os.path.exists(row_data):
+        if not isinstance(row_data, str):
+            # If the row data is an object, upload as a string
+            args[DataRow.row_data.name] = json.dumps(row_data)
+        elif os.path.exists(row_data):
+            # If row data is a local file path, upload it to server.
             args[DataRow.row_data.name] = self.client.upload_file(row_data)
         args[DataRow.dataset.name] = self
 
@@ -106,6 +109,7 @@ def convert_field_keys(items):
             mdo = self.client.get_data_row_metadata_ontology()
             args[DataRow.metadata_fields.name] = mdo.parse_upsert_metadata(
                 args[DataRow.metadata_fields.name])
+
         return self.client._create(DataRow, args)
 
     def create_data_rows_sync(self, items) -> None:

labelbox/schema/media_type.py

@@ -21,9 +21,9 @@ class MediaType(Enum):
 
     @classmethod
     def _missing_(cls, name):
-        """Handle missing null data types for projects 
+        """Handle missing null data types for projects
             created without setting allowedMediaType
-            Handle upper case names for compatibility with 
+            Handle upper case names for compatibility with
             the GraphQL"""
 
         if name is None: