fix: test

Signed-off-by: shuangkun <tsk2013uestc@163.com>
argoproj · Apr 12, 2024 · f016cba · f016cba
1 parent 0efe564
commit f016cba
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/workflow/controller/retry_tweak.go b/workflow/controller/retry_tweak.go
@@ -15,14 +15,13 @@ type RetryTweak = func(retryStrategy wfv1.RetryStrategy, nodes wfv1.Nodes, pod *
 func FindRetryNode(nodes wfv1.Nodes, nodeID string) *wfv1.NodeStatus {
 	boundaryID := nodes[nodeID].BoundaryID
 	boundaryNode := nodes[boundaryID]
-	templateName := boundaryNode.TemplateName
 	for _, node := range nodes {
 		if node.Type != wfv1.NodeTypeRetry {
 			continue
 		}
 		if boundaryID == "" && node.HasChild(nodeID) {
 			return &node
-		} else if boundaryNode.TemplateName != "" && node.TemplateName == templateName {
+		} else if boundaryNode.TemplateName != "" && node.TemplateName == boundaryNode.TemplateName {
 			return &node
 		} else if boundaryNode.TemplateRef != nil && node.TemplateRef != nil && node.TemplateRef.Name == boundaryNode.TemplateRef.Name && node.TemplateRef.Template == boundaryNode.TemplateRef.Template {
 			return &node