Fix task id when making the mapped task label in graph view (apache#28993)

bbovenzi · web-flow · commit c56887431a76 · 2023-01-17T16:36:59.000-03:00
diff --git a/airflow/www/static/js/graph.js b/airflow/www/static/js/graph.js
@@ -90,16 +90,16 @@ const updateNodeLabels = (node, instances) => {
   let { label } = node.value;
   // Check if there is a count of mapped instances
   if ((tasks[node.id] && tasks[node.id].is_mapped) || node.value.isMapped) {
-    const firstChildId = node.children[0].id;
+    const id = !!node.children && node.children.length ? node.children[0].id : node.value.id;
 
     let count = ' ';
 
     // get count from mapped_states or the first child's mapped_states
     // TODO: update this count for when we can nest mapped tasks inside of mapped task groups
     if (instances[node.id] && instances[node.id].mapped_states) {
       count = instances[node.id].mapped_states.length;
-    } else if (firstChildId && instances[firstChildId]) {
-      count = instances[firstChildId].mapped_states.length;
+    } else if (id && instances[id]) {
+      count = instances[id].mapped_states.length;
     }
 
     if (!label.includes(`[${count}]`)) {
