From b26ed4aa4dee395844531efa4a76a022183bec22 Mon Sep 17 00:00:00 2001
From: Isitha Subasinghe <isitha@pipekit.io>
Date: Thu, 31 Oct 2024 21:29:30 +1100
Subject: [PATCH] fix: use templateName where  possible else nothing (#13836)

Signed-off-by: isubasinghe <isitha@pipekit.io>
---
 ui/src/workflows/components/workflow-dag/workflow-dag.tsx | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/ui/src/workflows/components/workflow-dag/workflow-dag.tsx b/ui/src/workflows/components/workflow-dag/workflow-dag.tsx
index 3f2777aae00f..26865c1a5f9f 100644
--- a/ui/src/workflows/components/workflow-dag/workflow-dag.tsx
+++ b/ui/src/workflows/components/workflow-dag/workflow-dag.tsx
@@ -92,12 +92,12 @@ export class WorkflowDag extends React.Component<WorkflowDagProps, WorkflowDagRe
         );
     }
 
-    private nodeLabel(n: NodeStatus, parent?: NodeStatus) {
+    private nodeLabel(n: NodeStatus) {
         const phase = n.type === 'Suspend' && n.phase === 'Running' ? 'Suspended' : n.phase;
         let label = shortNodeName(n);
 
         if (this.state.showInvokingTemplateName) {
-            label = `${parent?.templateRef?.name ?? parent?.templateName ?? n.templateName}:${label}`;
+            label = n.templateName ? `${n.templateName}:${label}` : label;
         }
 
         return {
@@ -228,7 +228,7 @@ export class WorkflowDag extends React.Component<WorkflowDagProps, WorkflowDagRe
                 }
                 const isExpanded: boolean = this.state.expandNodes.has('*') || this.state.expandNodes.has(item.nodeName);
 
-                nodes.set(item.nodeName, this.nodeLabel(child, allNodes[item.parent]));
+                nodes.set(item.nodeName, this.nodeLabel(child));
                 edges.set({v: item.parent, w: item.nodeName}, {});
 
                 // If we have already considered the children of this node, don't consider them again