fix: apply MPI DependsOn feature on the latest commit change

Signed-off-by: Amir380-A <62997533+Amir380-A@users.noreply.github.com>

Author: amirialy

charts/kubeflow-trainer/files/runtimes/deepspeed-distributed.yaml

@@ -11,7 +11,6 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-
 apiVersion: trainer.kubeflow.org/v1alpha1
 kind: ClusterTrainingRuntime
 metadata:
@@ -39,6 +38,19 @@ spec:
         targetReplicatedJobs:
           - launcher
       replicatedJobs:
+        - name: launcher
+          template:
+            metadata:
+              labels:
+                trainer.kubeflow.org/trainjob-ancestor-step: trainer
+            spec:
+              template:
+                spec:
+                  containers:
+                    - name: node
+                      image: {{ include "trainer.runtimeImage" (list .Values.runtimes.deepspeedDistributed.image .) }}
+                      securityContext:
+                        runAsUser: 1000
         - name: node
           template:
             spec:
@@ -59,20 +71,3 @@ spec:
                         tcpSocket:
                           port: 2222
                         initialDelaySeconds: 5
-
-        - name: launcher
-          dependsOn:
-            - name: node
-              status: Ready
-          template:
-            metadata:
-              labels:
-                trainer.kubeflow.org/trainjob-ancestor-step: trainer
-            spec:
-              template:
-                spec:
-                  containers:
-                    - name: node
-                      image: {{ include "trainer.runtimeImage" (list .Values.runtimes.deepspeedDistributed.image .) }}
-                      securityContext:
-                        runAsUser: 1000

charts/kubeflow-trainer/files/runtimes/mlx-distributed.yaml

@@ -11,7 +11,6 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-
 apiVersion: trainer.kubeflow.org/v1alpha1
 kind: ClusterTrainingRuntime
 metadata:
@@ -39,6 +38,22 @@ spec:
         targetReplicatedJobs:
           - launcher
       replicatedJobs:
+        - name: launcher
+          dependsOn:
+            - name: node
+              status: Ready
+          template:
+            metadata:
+              labels:
+                trainer.kubeflow.org/trainjob-ancestor-step: trainer
+            spec:
+              template:
+                spec:
+                  containers:
+                    - name: node
+                      image: {{ include "trainer.runtimeImage" (list .Values.runtimes.mlxDistributed.image .) }}
+                      securityContext:
+                        runAsUser: 1000
         - name: node
           template:
             spec:
@@ -59,20 +74,3 @@ spec:
                         tcpSocket:
                           port: 2222
                         initialDelaySeconds: 5
-
-        - name: launcher
-          dependsOn:
-            - name: node
-              status: Ready
-          template:
-            metadata:
-              labels:
-                trainer.kubeflow.org/trainjob-ancestor-step: trainer
-            spec:
-              template:
-                spec:
-                  containers:
-                    - name: node
-                      image: {{ include "trainer.runtimeImage" (list .Values.runtimes.mlxDistributed.image .) }}
-                      securityContext:
-                        runAsUser: 1000