fix: removes USER var, adds IS_COLAB var

GoogleCloudPlatform · Jul 6, 2024 · 156647f · 156647f
1 parent e9ee305
commit 156647f
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/notebooks/official/pipelines/challenger_vs_blessed_deployment_method.ipynb b/notebooks/official/pipelines/challenger_vs_blessed_deployment_method.ipynb
@@ -168,13 +168,12 @@
       "outputs": [],
       "source": [
         "# Install the packages\n",
-        "USER=''\n",
-        "! pip3 install {USER} --upgrade google-cloud-aiplatform \\\n",
+        "! pip3 install --upgrade google-cloud-aiplatform \\\n",
         "                                'google-cloud-pipeline-components<2'\n",
-        "! pip3 install {USER}           tensorflow==2.5 \\\n",
-        "                                tensorflow_hub\n",
+        "! pip3 install tensorflow==2.5 \\\n",
+        "                tensorflow_hub\n",
         "  \n",
-        "! pip3 install {USER} --upgrade 'kfp<2'"
+        "! pip3 install --upgrade 'kfp<2'"
       ]
     },
     {
@@ -342,6 +341,10 @@
       },
       "outputs": [],
       "source": [
+        "import sys\n",
+        "\n",
+        "IS_COLAB = \"google.colab\" in sys.modules\n",
+        "\n",
         "if (\n",
         "    SERVICE_ACCOUNT == \"\"\n",
         "    or SERVICE_ACCOUNT is None\n",