nit

Signed-off-by: Kevin Su <[email protected]>
flyteorg · Oct 2, 2023 · fecb4cd · fecb4cd
1 parent 847f0d9
commit fecb4cd
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/flyteplugins/go/tasks/plugins/array/awsbatch/transformer.go b/flyteplugins/go/tasks/plugins/array/awsbatch/transformer.go
@@ -48,12 +48,12 @@ func FlyteTaskToBatchInput(ctx context.Context, tCtx pluginCore.TaskExecutionCon
 			"Required value not set, taskTemplate Container")
 	}
 
-	//jobConfig := newJobConfig().
-	//	MergeFromKeyValuePairs(taskTemplate.GetContainer().GetConfig()).
-	//	MergeFromConfigMap(tCtx.TaskExecutionMetadata().GetOverrides().GetConfig())
-	//if len(jobConfig.DynamicTaskQueue) == 0 {
-	//	return nil, errors.Errorf(errors.BadTaskSpecification, "config[%v] is missing", DynamicTaskQueueKey)
-	//}
+	jobConfig := newJobConfig().
+		MergeFromKeyValuePairs(taskTemplate.GetContainer().GetConfig()).
+		MergeFromConfigMap(tCtx.TaskExecutionMetadata().GetOverrides().GetConfig())
+	if len(jobConfig.DynamicTaskQueue) == 0 {
+		return nil, errors.Errorf(errors.BadTaskSpecification, "config[%v] is missing", DynamicTaskQueueKey)
+	}
 
 	inputReader := array.GetInputReader(tCtx, taskTemplate)
 	cmd, err := template.Render(
@@ -107,7 +107,7 @@ func FlyteTaskToBatchInput(ctx context.Context, tCtx pluginCore.TaskExecutionCon
 		}
 	}
 	submitJobInput.SetJobName(tCtx.TaskExecutionMetadata().GetTaskExecutionID().GetGeneratedName()).
-		SetJobDefinition(jobDefinition).SetJobQueue("tutorial").
+		SetJobDefinition(jobDefinition).SetJobQueue(jobConfig.DynamicTaskQueue).
 		SetRetryStrategy(toRetryStrategy(ctx, toBackoffLimit(taskTemplate.Metadata), cfg.MinRetries, cfg.MaxRetries)).
 		SetContainerOverrides(toContainerOverrides(ctx, append(cmd, args...), &resources, envVars)).
 		SetTimeout(toTimeout(taskTemplate.Metadata.GetTimeout(), cfg.DefaultTimeOut.Duration))