Prevent tool from crashing if the classification is missing attributes.

responsible-ai-collaborative · Sep 14, 2023 · 1424355 · 1424355
1 parent e5dd230
commit 1424355
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/site/gatsby-site/src/pages/apps/csettool/{mongodbAiidprodIncidents.incident_id}.js b/site/gatsby-site/src/pages/apps/csettool/{mongodbAiidprodIncidents.incident_id}.js
@@ -34,13 +34,15 @@ const ToolPage = (props) => {
         };
 
         for (const classification of data.classifications) {
-          const json = classification.attributes.find(
-            (a) => a.short_name == attribute.short_name
-          ).value_json;
+          const item = classification.attributes.find((a) => a.short_name == attribute.short_name);
 
-          const value = JSON.parse(json);
+          if (item) {
+            const value = JSON.parse(item.value_json);
 
-          row[classification.namespace] = value;
+            row[classification.namespace] = value;
+          } else {
+            row[classification.namespace] = null;
+          }
         }
 
         rows.push(row);