moving fn to be on attachment instead

Author: jtsodapop

labelbox/schema/asset_attachment.py

@@ -1,11 +1,11 @@
 from enum import Enum
 from typing import Dict
 
-from labelbox.orm.db_object import DbObject
+from labelbox.orm.db_object import DbObject, Deletable
 from labelbox.orm.model import Field
 
 
-class AssetAttachment(DbObject):
+class AssetAttachment(DbObject, Deletable):
     """ Asset attachment provides extra context about an asset while labeling.
 
     Attributes:
@@ -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,12 +95,4 @@ def create_attachment(self, attachment_type,
                 data_row_id_param: self.uid
             })
         return Entity.AssetAttachment(self.client,
-                                      res["createDataRowAttachment"])
-
-    def delete_attachment(self, attachment_id: str) -> 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": attachment_id})
+                                      res["createDataRowAttachment"])