Merge pull request #485 from Labelbox/AL-1716

[AL-1716] Datarow Delete Attachment

Author: jtsodapop

labelbox/schema/asset_attachment.py

@@ -42,3 +42,11 @@ def validate_attachment_type(cls, attachment_type: str) -> None:
             raise ValueError(
                 f"meta_type must be one of {valid_types}. Found {attachment_type}"
             )
+
+    def delete(self) -> None:
+        """Deletes an attachment on the data row."""
+        query_str = """mutation deleteDataRowAttachmentPyApi($attachment_id: ID!) {
+            deleteDataRowAttachment(where: {id: $attachment_id}) {
+                    id}
+            }"""
+        self.client.execute(query_str, {"attachment_id": self.uid})

labelbox/schema/data_row.py

@@ -95,4 +95,4 @@ def create_attachment(self, attachment_type,
                 data_row_id_param: self.uid
             })
         return Entity.AssetAttachment(self.client,
-                                      res["createDataRowAttachment"])
+                                      res["createDataRowAttachment"])

tests/integration/test_data_rows.py

@@ -6,7 +6,6 @@
 import requests
 
 from labelbox import DataRow
-from labelbox.exceptions import InvalidQueryError
 
 
 def test_get_data_row(datarow, client):
@@ -276,3 +275,17 @@ def test_create_data_rows_sync_mixed_upload(dataset, image_url):
             DataRow.row_data: image_url
         }] * n_urls + [fp.name] * n_local)
     assert len(list(dataset.data_rows())) == n_local + n_urls
+
+
+def test_delete_data_row_attachment(datarow, image_url):
+    attachments = []
+    to_attach = [("IMAGE", image_url), ("TEXT", "test-text"),
+                 ("IMAGE_OVERLAY", image_url), ("HTML", image_url)]
+    for attachment_type, attachment_value in to_attach:
+        attachments.append(
+            datarow.create_attachment(attachment_type, attachment_value))
+
+    for attachment in attachments:
+        attachment.delete()
+
+    assert len(list(datarow.attachments())) == 0