fix: update risk matrix to use ref_id instead of rid

intuitem · Dec 5, 2024 · 3f351d8 · 3f351d8
1 parent 60f5495
commit 3f351d8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/backend/core/helpers.py b/backend/core/helpers.py
@@ -1049,9 +1049,9 @@ def build_scenario_clusters(risk_assessment: RiskAssessment):
         "created_at"
     ):
         if ri.current_level >= 0:
-            risk_matrix_current[ri.current_proba][ri.current_impact].add(ri.rid)
+            risk_matrix_current[ri.current_proba][ri.current_impact].add(ri.ref_id)
         if ri.residual_level >= 0:
-            risk_matrix_residual[ri.residual_proba][ri.residual_impact].add(ri.rid)
+            risk_matrix_residual[ri.residual_proba][ri.residual_impact].add(ri.ref_id)
 
     return {"current": risk_matrix_current, "residual": risk_matrix_residual}