moving restartConfig for auto resume policy to under Worker key to align with kubeflow docs

Author: rnadimp

launcher/nemo/k8s_templates/training/training.yaml

@@ -25,6 +25,7 @@ spec:
   pytorchReplicaSpecs:
     Worker:
       replicas: {{ $config.nodes }}
+      restartPolicy: {{ $config.restartPolicy }}
       template:
         {{- if $config.customLabels }}
         metadata:
@@ -101,7 +102,6 @@ spec:
             - mountPath: /var/log/aws/clusters
               name: aws-clusters-logs
               readOnly: true
-          restartPolicy: {{ $config.restartPolicy }}
 
           {{- if (or $config.labelSelector.required $config.labelSelector.preferred) }}
           affinity:

tests/k8s_workflow/k8s_baseline_artifacts/llama-8b/k8s_template/templates/training.yaml

@@ -25,6 +25,7 @@ spec:
   pytorchReplicaSpecs:
     Worker:
       replicas: {{ $config.nodes }}
+      restartPolicy: {{ $config.restartPolicy }}
       template:
         {{- if $config.customLabels }}
         metadata:
@@ -101,7 +102,6 @@ spec:
             - mountPath: /var/log/aws/clusters
               name: aws-clusters-logs
               readOnly: true
-          restartPolicy: {{ $config.restartPolicy }}
 
           {{- if (or $config.labelSelector.required $config.labelSelector.preferred) }}
           affinity:

tests/k8s_workflow/k8s_baseline_artifacts/test_custom/k8s_template/templates/training.yaml

@@ -25,6 +25,7 @@ spec:
   pytorchReplicaSpecs:
     Worker:
       replicas: {{ $config.nodes }}
+      restartPolicy: {{ $config.restartPolicy }}
       template:
         {{- if $config.customLabels }}
         metadata:
@@ -101,7 +102,6 @@ spec:
             - mountPath: /var/log/aws/clusters
               name: aws-clusters-logs
               readOnly: true
-          restartPolicy: {{ $config.restartPolicy }}
 
           {{- if (or $config.labelSelector.required $config.labelSelector.preferred) }}
           affinity: