ai_reports/serializer: use get() for label to prevent errors for unknown keys

Julian Dehm · m4ra · commit ca3aac54dbbe · 2024-09-19T16:15:24.000+03:00
diff --git a/apps/ai_reports/serializers.py b/apps/ai_reports/serializers.py
@@ -25,7 +25,7 @@ class Meta:
     def get_label(self, ai_report: AiReport):
         """Replace label codes with description and remove catneutral and catnodecis"""
         return [
-            (label, settings.XAI_LABELS[label])
+            (label, settings.XAI_LABELS.get(label, label))
             for label in ai_report.label
             if label != "catneutral" and label != "catnodecis"
         ]

Original file line number	Diff line number	Diff line change
`@@ -25,7 +25,7 @@ class Meta:`
`25`	`25`	`def get_label(self, ai_report: AiReport):`
`26`	`26`	`"""Replace label codes with description and remove catneutral and catnodecis"""`
`27`	`27`	`return [`
`28`		`- (label, settings.XAI_LABELS[label])`
	`28`	`+ (label, settings.XAI_LABELS.get(label, label))`
`29`	`29`	`for label in ai_report.label`
`30`	`30`	`if label != "catneutral" and label != "catnodecis"`
`31`	`31`	`]`