Build - Fix building TF images (#2736)

.cloudbuild.yaml

@@ -100,6 +100,11 @@ steps:
   args: ['-c', 'cd /workspace/components/kubeflow/launcher && ./build_image.sh -p $PROJECT_ID -t $COMMIT_SHA']
   id: 'buildLauncher'
   waitFor: ["-"]
+- id: 'buildCpuTrainer'
+  name: 'gcr.io/cloud-builders/docker'
+  entrypoint: '/bin/bash'
+  args: ['-c', 'cd /workspace/components/kubeflow/dnntrainer && ./build_image.sh -p $PROJECT_ID -t $COMMIT_SHA -l ml-pipeline-kubeflow-tf-trainer -b 1.6.0']
+  waitFor: ["-"]
 - name: 'gcr.io/cloud-builders/docker'
   entrypoint: '/bin/bash'
   args: ['-c', 'cd /workspace/components/kubeflow/dnntrainer && ./build_image.sh -p $PROJECT_ID -t $COMMIT_SHA -l ml-pipeline-kubeflow-tf-trainer-gpu -b 1.6.0-gpu']
@@ -177,7 +182,7 @@ images:
 - 'gcr.io/$PROJECT_ID/ml-pipeline-kubeflow-deployer:$COMMIT_SHA'
 - 'gcr.io/$PROJECT_ID/ml-pipeline-kubeflow-tf-trainer:$COMMIT_SHA'
 - 'gcr.io/$PROJECT_ID/ml-pipeline-kubeflow-tf-trainer-gpu:$COMMIT_SHA'
-- 'gcr.io/$PROJECT_ID/ml-pipeline-kubeflow-tf:$COMMIT_SHA'
+- 'gcr.io/$PROJECT_ID/ml-pipeline-kubeflow-tfjob:$COMMIT_SHA'
 
 # Images for the Dataproc-based pipeline components
 - 'gcr.io/$PROJECT_ID/ml-pipeline-dataproc-analyze:$COMMIT_SHA'

components/release.sh

@@ -27,7 +27,7 @@ images=(
   "ml-pipeline-kubeflow-deployer"
   "ml-pipeline-kubeflow-tf-trainer"
   "ml-pipeline-kubeflow-tf-trainer-gpu"
-  "ml-pipeline-kubeflow-tf"
+  "ml-pipeline-kubeflow-tfjob"
   "ml-pipeline-dataproc-analyze"
   "ml-pipeline-dataproc-create-cluster"
   "ml-pipeline-dataproc-delete-cluster"