better

Signed-off-by: Katrina Rogan <[email protected]>
flyteorg · Jul 22, 2024 · f398c50 · f398c50
1 parent 947331a
commit f398c50
Showing 1 changed file with 1 addition and 12 deletions.
diff --git a/flyteadmin/pkg/manager/impl/util/single_task_execution.go b/flyteadmin/pkg/manager/impl/util/single_task_execution.go
@@ -22,10 +22,6 @@ import (
 
 const maxNodeIDLength = 63
 
-var defaultRetryStrategy = core.RetryStrategy{
-	Retries: 3,
-}
-
 const systemNamePrefix = ".flytegen.%s"
 
 const noInputNodeID = ""
@@ -85,13 +81,6 @@ func CreateOrGetWorkflowModel(
 		Version: workflowIdentifier.Version,
 	})
 
-	var retryStrategy *core.RetryStrategy
-	if task.GetClosure().GetCompiledTask().GetTemplate().GetMetadata().GetRetries().GetRetries() > 0 {
-		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
@@ -106,7 +95,7 @@ func CreateOrGetWorkflowModel(
 						Id: generateNodeNameFromTask(taskIdentifier.Name),
 						Metadata: &core.NodeMetadata{
 							Name:    generateNodeNameFromTask(taskIdentifier.Name),
-							Retries: retryStrategy,
+							Retries: task.GetClosure().GetCompiledTask().GetTemplate().GetMetadata().GetRetries(),
 						},
 						Inputs: generateBindings(*task.Closure.CompiledTask.Template.Interface.Inputs, noInputNodeID),
 						Target: &core.Node_TaskNode{