more changes

Signed-off-by: Yuvraj <[email protected]>
flyteorg · Aug 27, 2021 · 3315a3a · 3315a3a
1 parent c6746ff
commit 3315a3a
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/go/tasks/plugins/k8s/kfoperators/mpi/mpi.go b/go/tasks/plugins/k8s/kfoperators/mpi/mpi.go
@@ -65,9 +65,14 @@ func (mpiOperatorResourceHandler) BuildResource(ctx context.Context, taskCtx plu
 	launcherReplicas := mpiTaskExtraArgs.GetNumLauncherReplicas()
 	slots := mpiTaskExtraArgs.GetSlots()
 
+	workersPodSpec := podSpec.DeepCopy()
+	for _,container := range workersPodSpec.Containers {
+		container.Args = []string{}
+		container.Command = []string{}
+	}
+
 	jobSpec := mpi.MPIJobSpec{
 		SlotsPerWorker: &slots,
-		//MainContainer: "",
 		MPIReplicaSpecs: map[mpi.MPIReplicaType]*commonKf.ReplicaSpec{
 			mpi.MPIReplicaTypeLauncher: &commonKf.ReplicaSpec{
 				Replicas: &launcherReplicas,
@@ -79,7 +84,7 @@ func (mpiOperatorResourceHandler) BuildResource(ctx context.Context, taskCtx plu
 			mpi.MPIReplicaTypeWorker: &commonKf.ReplicaSpec{
 				Replicas: &workers,
 				Template: v1.PodTemplateSpec{
-					Spec: *podSpec,
+					Spec: *workersPodSpec,
 				},
 				RestartPolicy: commonKf.RestartPolicyNever,
 			},