meh

Signed-off-by: Katrina Rogan <[email protected]>
flyteorg · Jul 22, 2024 · 3f0d570 · 3f0d570
1 parent f398c50
commit 3f0d570
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/flyteadmin/pkg/manager/impl/util/single_task_execution.go b/flyteadmin/pkg/manager/impl/util/single_task_execution.go
@@ -22,6 +22,10 @@ import (
 
 const maxNodeIDLength = 63
 
+var defaultRetryStrategy = core.RetryStrategy{
+	Retries: 3,
+}
+
 const systemNamePrefix = ".flytegen.%s"
 
 const noInputNodeID = ""
@@ -81,6 +85,15 @@ func CreateOrGetWorkflowModel(
 		Version: workflowIdentifier.Version,
 	})
 
+	var retryStrategy *core.RetryStrategy
+	if task.GetClosure().GetCompiledTask().GetTemplate() != nil &&
+		task.GetClosure().GetCompiledTask().GetTemplate().GetMetadata() != nil &&
+		task.GetClosure().GetCompiledTask().GetTemplate().GetMetadata().GetRetries() != nil {
+		retryStrategy = task.GetClosure().GetCompiledTask().GetTemplate().GetMetadata().GetRetries()
+	} else {
+		retryStrategy = &defaultRetryStrategy
+	}
+
 	if err != nil {
 		if ferr, ok := err.(errors.FlyteAdminError); !ok || ferr.Code() != codes.NotFound {
 			return nil, err
@@ -95,7 +108,7 @@ func CreateOrGetWorkflowModel(
 						Id: generateNodeNameFromTask(taskIdentifier.Name),
 						Metadata: &core.NodeMetadata{
 							Name:    generateNodeNameFromTask(taskIdentifier.Name),
-							Retries: task.GetClosure().GetCompiledTask().GetTemplate().GetMetadata().GetRetries(),
+							Retries: retryStrategy,
 						},
 						Inputs: generateBindings(*task.Closure.CompiledTask.Template.Interface.Inputs, noInputNodeID),
 						Target: &core.Node_TaskNode{