Allow removing a csp estimate from a task order

2019-01-22 13:05:04 -05:00
parent c155de0e84
commit 672c562323
2 changed files with 10 additions and 0 deletions
--- a/atst/models/task_order.py
+++ b/atst/models/task_order.py
@@ -86,6 +86,8 @@ class TaskOrder(Base, mixins.TimestampsMixin):
            self._csp_estimate = Attachment.attach(
                new_csp_estimate, "task_order", self.id
            )
+        elif not new_csp_estimate and self._csp_estimate:
+            self._csp_estimate = None
        else:
            raise TypeError("Could not set csp_estimate with invalid type")