ds del column

DagsHub · Jan 17, 2024 · 292c1ca · 292c1ca
1 parent 71b6a99
commit 292c1ca
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/dagshub/data_engine/client/models.py b/dagshub/data_engine/client/models.py
@@ -61,6 +61,7 @@ class MetadataFieldSchema:
     valueType: MetadataFieldType = field(metadata=config(encoder=lambda val: val.value))
     multiple: bool
     tags: Optional[Set[str]]
+    remove: Optional[bool]
 
     def __repr__(self):
         res = f"{self.name} ({self.valueType.value})"

diff --git a/dagshub/data_engine/model/metadata_field_builder.py b/dagshub/data_engine/model/metadata_field_builder.py
@@ -46,6 +46,10 @@ def schema(self) -> MetadataFieldSchema:
             )
         return self._schema
 
+    def delete(self):
+        self._schema.remove = True
+        return self
+
     def set_type(self, t: Union[Type, DagshubDataType]) -> "MetadataFieldBuilder":
         """
         Set the type of the field.