SchedulerName config (flyteorg#78)

Add SchedulerName option that will propagate to PodSpec

flyteplugins/go/tasks/pluginmachinery/flytek8s/config/config.go

@@ -49,6 +49,8 @@ type K8sPluginConfig struct {
 	// Node Selector Labels for interruptible pods: Similar to InterruptibleTolerations, these node selector labels are added for pods that can tolerate
 	// eviction.
 	InterruptibleNodeSelector map[string]string `json:"interruptible-node-selector" pflag:"-,Defines a set of node selector labels to add to the interruptible pods."`
+	// Scheduler name.
+	SchedulerName string `json:"scheduler-name" pflag:",Defines scheduler name."`
 }
 
 // Retrieves the current k8s plugin config or default.

flyteplugins/go/tasks/pluginmachinery/flytek8s/pod_helper.go

@@ -43,6 +43,7 @@ func ToK8sPodSpec(ctx context.Context, taskExecutionMetadata pluginsCore.TaskExe
 			Tolerations:        GetPodTolerations(taskExecutionMetadata.IsInterruptible(), c.Resources),
 			ServiceAccountName: taskExecutionMetadata.GetK8sServiceAccount(),
 			NodeSelector:       config.GetK8sPluginConfig().InterruptibleNodeSelector,
+			SchedulerName:      config.GetK8sPluginConfig().SchedulerName,
 		}, nil
 	}
 	return &v1.PodSpec{
@@ -51,6 +52,7 @@ func ToK8sPodSpec(ctx context.Context, taskExecutionMetadata pluginsCore.TaskExe
 		Containers:         containers,
 		Tolerations:        GetPodTolerations(taskExecutionMetadata.IsInterruptible(), c.Resources),
 		ServiceAccountName: taskExecutionMetadata.GetK8sServiceAccount(),
+		SchedulerName:      config.GetK8sPluginConfig().SchedulerName,
 	}, nil
 
 }