ADD: media attributes

Author: gdj0nes

labelbox/schema/data_row.py

@@ -13,6 +13,7 @@ class DataRow(DbObject, Updateable, BulkDeletable):
             Otherwise, it's treated as an external URL.
         updated_at (datetime)
         created_at (datetime)
+        media_attributes (dict): generated media attributes for the datarow
 
         dataset (Relationship): `ToOne` relationship to Dataset
         created_by (Relationship): `ToOne` relationship to User
@@ -25,13 +26,15 @@ class DataRow(DbObject, Updateable, BulkDeletable):
     row_data = Field.String("row_data")
     updated_at = Field.DateTime("updated_at")
     created_at = Field.DateTime("created_at")
+    media_attributes = Field.Json("media_attributes")
 
     # Relationships
     dataset = Relationship.ToOne("Dataset")
     created_by = Relationship.ToOne("User", False, "created_by")
     organization = Relationship.ToOne("Organization", False)
     labels = Relationship.ToMany("Label", True)
     metadata = Relationship.ToMany("AssetMetadata", False, "metadata")
+
     predictions = Relationship.ToMany("Prediction", False)
 
     supported_meta_types = {
@@ -90,3 +93,4 @@ def create_metadata(self, meta_type, meta_value):
                 data_row_id_param: self.uid
             })
         return Entity.AssetMetadata(self.client, res["createAssetMetadata"])
+

tests/integration/test_data_rows.py

@@ -77,6 +77,7 @@ def test_data_row_single_creation(dataset, rand_gen):
     assert data_row.organization() == client.get_organization()
     assert requests.get(IMG_URL).content == \
         requests.get(data_row.row_data).content
+    assert data_row.media_attributes is not None
 
     with NamedTemporaryFile() as fp:
         data = rand_gen(str).encode()