missing setting IS_COLAB (#2288)

2026-09-26 14:42:04 +00:00 · 2023-09-29 15:54:51 +00:00
parent 6726b90892
commit 4e3fbed9f7
1 changed files with 3 additions and 0 deletions
@@ -438,6 +438,9 @@
        "    or SERVICE_ACCOUNT is None\n",
        "    or SERVICE_ACCOUNT == \"[your-service-account]\"\n",
        "):\n",
+        "    import sys\n",
+        "    IS_COLAB = 'google.colab' in sys.modules\n",
+        "\n",
        "    # Get your service account from gcloud\n",
        "    if not IS_COLAB:\n",
        "        shell_output = !gcloud auth list 2>/dev/null\n",