Do not remove the cost model when updating the incident. (Netflix#4567)

metroid-samus · web-flow · commit 1d405ae1fa0b · 2024-04-02T09:12:21.000-07:00
diff --git a/src/dispatch/incident/service.py b/src/dispatch/incident/service.py
@@ -4,6 +4,7 @@
     :copyright: (c) 2019 by Netflix Inc., see AUTHORS for more
     :license: Apache, see LICENSE for more details.
 """
+
 import logging
 
 from datetime import datetime, timedelta
@@ -338,12 +339,14 @@ def update(*, db_session, incident: Incident, incident_in: IncidentUpdate) -> In
             incident_priority_in=incident_in.incident_priority,
         )
 
-    cost_model = None
-    if incident_in.cost_model and incident_in.cost_model.id != incident.cost_model_id:
-        cost_model = cost_model_service.get_cost_model_by_id(
+    cost_model = (
+        cost_model_service.get_cost_model_by_id(
             db_session=db_session,
             cost_model_id=incident_in.cost_model.id,
         )
+        if incident_in.cost_model
+        else None
+    )
 
     cases = []
     for c in incident_in.cases:
