addition of media attributes and metadata for a Data object

Author: jtsodapop

labelbox/data/annotation_types/data/base_data.py

@@ -1,5 +1,5 @@
 from abc import ABC
-from typing import Optional
+from typing import Optional, Dict, List, Any
 
 from pydantic import BaseModel
 
@@ -11,3 +11,5 @@ class BaseData(BaseModel, ABC):
     """
     external_id: Optional[str] = None
     uid: Optional[str] = None
+    media_attributes: Optional[Dict[str, Any]] = None
+    metadata: Optional[List[Dict[str, Any]]]

labelbox/data/serialization/labelbox_v1/label.py

@@ -1,6 +1,6 @@
 from labelbox.data.annotation_types.data.tiled_image import TiledImageData
 from labelbox.utils import camel_case
-from typing import List, Optional, Union
+from typing import List, Optional, Union, Dict, Any
 
 from pydantic import BaseModel, Field
 
@@ -126,6 +126,10 @@ class LBV1Label(BaseModel):
     row_data: str = Field(None, alias="Labeled Data")
     id: Optional[str] = Field(None, alias='ID')
     external_id: Optional[str] = Field(None, alias="External ID")
+    data_row_media_attributes: Optional[Dict[str, Any]] = Field(
+        ..., alias="Media Attributes")
+    data_row_metadata: Optional[List[Dict[str, Any]]] = Field(
+        ..., alias="DataRow Metadata")
 
     created_by: Optional[str] = Extra('Created By')
     project_name: Optional[str] = Extra('Project Name')
@@ -181,7 +185,9 @@ def _data_row_to_common(
         data_row_info = {
             'url' if self._is_url() else 'text': self.row_data,
             'external_id': self.external_id,
-            'uid': self.data_row_id
+            'uid': self.data_row_id,
+            'media_attributes': self.data_row_media_attributes,
+            'metadata': self.data_row_metadata
         }
 
         self.media_type = self.media_type or self._infer_media_type()