Merge pull request GoogleCloudPlatform#162 from GoogleCloudPlatform/ajayhemnani-patch-2

Update multiple_frameworks_lab.ipynb

Author: ajayhemnani

continuous_training/kubeflow/labs/multiple_frameworks_lab.ipynb

@@ -22,6 +22,22 @@
     "**PRIOR TO STARTING THE LAB:** Make sure you create a new instance with AI Platform Pipelines. Once the GKE cluster is spun up, copy the endpoint because you will need it in this lab. "
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "Install needed package"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "%pip install --upgrade google-cloud-bigquery-storage"
+   ]
+  },
   {
    "cell_type": "markdown",
    "metadata": {},
@@ -1098,7 +1114,7 @@
     "\n",
     "\n",
     "\n",
-    "**NOTE that your pipeline run may fail due to the bug in a BigQuery component that does not handle certain race conditions. If you observe the pipeline failure, submit another pipeline run using the KFP UI using the steps above**\n",
+    "**NOTE that your pipeline run may fail due to the bug in a BigQuery component that does not handle certain race conditions. If you observe the pipeline failure, please click on Retry on the top right in the KFP UI to re-run the failed steps.**\n",
     "\n",
     "\n",
     "Now let's set up a recurring run:\n",