Set restart policy

kubeflow · Mar 22, 2018 · a375ae5 · a375ae5
1 parent e438720
commit a375ae5
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/pkg/controller/controller_pod.go b/pkg/controller/controller_pod.go
@@ -108,6 +108,11 @@ func (tc *TFJobController) reconcilePods(
 				})
 			}
 
+			// Set restart policy
+			if spec.RestartPolicy != "" && spec.RestartPolicy != "ExitCode" {
+				pTemplate.Spec.RestartPolicy = v1.RestartPolicy(spec.RestartPolicy)
+			}
+
 			err := tc.podControl.CreatePodsWithControllerRef(tfjob.Namespace, pTemplate, tfjob, controllerRef)
 			if err != nil && errors.IsTimeout(err) {
 				// Pod is created but its initialization has timed out.