add user-gcp-sa annotation to 'train' steps (amygdala#31)

Author: amygdala

ml/kubeflow-pipelines/samples/kubeflow-tf/gh_summ.py

@@ -14,6 +14,8 @@
 
 
 import kfp.dsl as dsl
+import kfp.gcp as gcp
+
 
 @dsl.pipeline(
   name='Github issue summarization',
@@ -38,7 +40,7 @@ def gh_summ(
           "--train-steps", train_steps, "--deploy-webapp" , deploy_webapp],
       file_outputs={'output': '/tmp/output'}
 
-      )
+      ).apply(gcp.use_gcp_secret('user-gcp-sa'))
 
   serve = dsl.ContainerOp(
       name = 'serve',

ml/kubeflow-pipelines/samples/kubeflow-tf/gh_summ.py.tar.gz

@@ -136,7 +136,7 @@
     "          \"--train-steps\", train_steps, \"--deploy-webapp\" , deploy_webapp],\n",
     "      file_outputs={'output': '/tmp/output'}\n",
     "\n",
-    "      )\n",
+    "      ).apply(gcp.use_gcp_secret('user-gcp-sa'))\n",
     "\n",
     "  serve = dsl.ContainerOp(\n",
     "      name = 'serve',\n",
@@ -275,7 +275,7 @@
     "          \"--train-steps\", train_steps, \"--deploy-webapp\" , deploy_webapp],\n",
     "      file_outputs={'output': '/tmp/output'}\n",
     "\n",
-    "      )\n",
+    "      ).apply(gcp.use_gcp_secret('user-gcp-sa'))\n",
     "  train.after(preproc)\n",
     "\n",
     "  serve = dsl.ContainerOp(\n",