Cleaned up cleanup section

2026-09-26 14:42:04 +00:00 · 2022-02-15 14:22:42 -05:00
1 changed files with 1 additions and 4 deletions
@@ -675,9 +675,6 @@
      },
      "outputs": [],
      "source": [
-        "# Warning: Setting this to true will delete everything in your bucket\n",
-        "delete_bucket = False\n",
-        "\n",
        "# Delete endpoint resource\n",
        "! gcloud ai endpoints delete $ENDPOINT_NAME --quiet --region $REGION_NAME\n",
        "\n",
@@ -687,7 +684,7 @@
        "# Delete Cloud Storage objects that were created\n",
        "! gsutil -m rm -r $JOB_DIR\n",
        "\n",
-        "if delete_bucket and \"BUCKET_URI\" in globals():\n",
+        "if os.getenv(\"IS_TESTING\"):\n",
        "    ! gsutil -m rm -r $BUCKET_URI"
      ]
    }