Use latest Tensorflow image for Tensorboard. (#2140)

* Use latest Tensorflow image for Tensorboard. * Fix tests
kubeflow · Sep 21, 2019 · d5316f0 · d5316f0
1 parent ab8f5ed
commit d5316f0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/backend/src/crd/controller/viewer/reconciler/reconciler.go b/backend/src/crd/controller/viewer/reconciler/reconciler.go
@@ -165,7 +165,7 @@ func setPodSpecForTensorboard(view *viewerV1beta1.Viewer, s *corev1.PodSpec) {
 
 	c := &s.Containers[0]
 	c.Name = view.Name + "-pod"
-	c.Image = "tensorflow/tensorflow:1.11.0"
+	c.Image = "tensorflow/tensorflow"
 	c.Args = []string{
 		"tensorboard",
 		fmt.Sprintf("--logdir=%s", view.Spec.TensorboardSpec.LogDir),

diff --git a/backend/src/crd/controller/viewer/reconciler/reconciler_test.go b/backend/src/crd/controller/viewer/reconciler/reconciler_test.go
@@ -174,7 +174,7 @@ func TestReconcile_EachViewerCreatesADeployment(t *testing.T) {
 				Spec: corev1.PodSpec{
 					Containers: []corev1.Container{{
 						Name:  "viewer-123-pod",
-						Image: "tensorflow/tensorflow:1.11.0",
+						Image: "tensorflow/tensorflow",
 						Args: []string{
 							"tensorboard",
 							"--logdir=gs://tensorboard/logdir",
@@ -271,7 +271,7 @@ func TestReconcile_ViewerUsesSpecifiedVolumeMountsForDeployment(t *testing.T) {
 				Spec: corev1.PodSpec{
 					Containers: []corev1.Container{{
 						Name:  "viewer-123-pod",
-						Image: "tensorflow/tensorflow:1.11.0",
+						Image: "tensorflow/tensorflow",
 						Args: []string{
 							"tensorboard",
 							"--logdir=gs://tensorboard/logdir",