Add template to replicaSpec for examples

kubeflow · Jan 30, 2018 · 16f5d86 · 16f5d86
1 parent e3146c4
commit 16f5d86
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/examples/tf_job.yaml b/examples/tf_job.yaml
@@ -22,3 +22,9 @@ spec:
           restartPolicy: OnFailure
     - replicas: 2
       tfReplicaType: PS
+      template:
+        spec:
+          containers:
+            - image: gcr.io/tf-on-k8s-dogfood/tf_sample:dc944ff
+              name: tensorflow
+          restartPolicy: OnFailure
diff --git a/test/e2e/simple_job.yaml.template b/test/e2e/simple_job.yaml.template
@@ -22,3 +22,9 @@ spec:
           restartPolicy: OnFailure
     - replicas: 2
       tfReplicaType: PS
+      template:
+        spec:
+          containers:
+            - image: gcr.io/tf-on-k8s-dogfood/tf_sample:dc944ff
+              name: tensorflow
+          restartPolicy: OnFailure