Add uuid to reduce bucket name crashing (#2894)

* Add uuid to notebooks. * Lint
GoogleCloudPlatform · Apr 23, 2024 · 6c3bd0e · 6c3bd0e
1 parent cff2942
commit 6c3bd0e
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 4 deletions.
diff --git a/notebooks/community/model_garden/model_garden_gemma_finetuning_on_vertex.ipynb b/notebooks/community/model_garden/model_garden_gemma_finetuning_on_vertex.ipynb
@@ -100,6 +100,7 @@
         "# @markdown **[Optional]** Set the GCS BUCKET_URI to store the experiment artifacts, if you want to use your own bucket. **If not set, a unique GCS bucket will be created automatically on your behalf**.\n",
         "\n",
         "import os\n",
+        "import uuid\n",
         "from datetime import datetime\n",
         "from typing import Tuple\n",
         "\n",
@@ -118,7 +119,7 @@
         "BUCKET_URI = \"gs://\"  # @param {type:\"string\"}\n",
         "if not BUCKET_URI.strip() or BUCKET_URI == \"gs://\":\n",
         "    # Create a unique GCS bucket for this notebook if not specified\n",
-        "    BUCKET_URI = f\"gs://{PROJECT_ID}-tmp-{now}\"\n",
+        "    BUCKET_URI = f\"gs://{PROJECT_ID}-tmp-{now}-{str(uuid.uuid4())[:4]}\"\n",
         "    ! gsutil mb -l {REGION} {BUCKET_URI}\n",
         "assert BUCKET_URI.startswith(\"gs://\"), \"BUCKET_URI must start with `gs://`.\"\n",
         "\n",

diff --git a/notebooks/community/model_garden/model_garden_pytorch_gemma_peft_finetuning_hf.ipynb b/notebooks/community/model_garden/model_garden_pytorch_gemma_peft_finetuning_hf.ipynb
@@ -101,6 +101,7 @@
         "\n",
         "import os\n",
         "import sys\n",
+        "import uuid\n",
         "from datetime import datetime\n",
         "from typing import Tuple\n",
         "\n",
@@ -119,7 +120,7 @@
         "BUCKET_URI = \"gs://\"  # @param {type:\"string\"}\n",
         "if not BUCKET_URI.strip() or BUCKET_URI == \"gs://\":\n",
         "    # Create a unique GCS bucket for this notebook if not specified\n",
-        "    BUCKET_URI = f\"gs://{PROJECT_ID}-tmp-{now}\"\n",
+        "    BUCKET_URI = f\"gs://{PROJECT_ID}-tmp-{now}-{str(uuid.uuid4())[:4]}\"\n",
         "    ! gsutil mb -l {REGION} {BUCKET_URI}\n",
         "assert BUCKET_URI.startswith(\"gs://\"), \"BUCKET_URI must start with `gs://`.\"\n",
         "\n",

diff --git a/notebooks/community/model_garden/model_garden_pytorch_llama2_deployment.ipynb b/notebooks/community/model_garden/model_garden_pytorch_llama2_deployment.ipynb
@@ -97,6 +97,7 @@
         "\n",
         "# Import the necessary packages\n",
         "import os\n",
+        "import uuid\n",
         "from datetime import datetime\n",
         "from typing import Tuple\n",
         "\n",
@@ -119,7 +120,7 @@
         "BUCKET_URI = \"gs://\"  # @param {type:\"string\"}\n",
         "\n",
         "if BUCKET_URI is None or BUCKET_URI.strip() == \"\" or BUCKET_URI == \"gs://\":\n",
-        "    BUCKET_URI = f\"gs://{PROJECT_ID}-tmp-{now}\"\n",
+        "    BUCKET_URI = f\"gs://{PROJECT_ID}-tmp-{now}-{str(uuid.uuid4())[:4]}\"\n",
         "    ! gsutil mb -l {REGION} {BUCKET_URI}\n",
         "else:\n",
         "    assert BUCKET_URI.startswith(\"gs://\"), \"BUCKET_URI must start with `gs://`.\"\n",

diff --git a/notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_xl_1_0.ipynb b/notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_xl_1_0.ipynb
@@ -98,6 +98,7 @@
         "import base64\n",
         "import os\n",
         "import sys\n",
+        "import uuid\n",
         "from datetime import datetime\n",
         "from io import BytesIO\n",
         "\n",
@@ -122,7 +123,7 @@
         "\n",
         "# Create a unique GCS bucket for this notebook, if not specified by the user.\n",
         "if BUCKET_URI is None or BUCKET_URI.strip() == \"\" or BUCKET_URI == \"gs://\":\n",
-        "    BUCKET_URI = f\"gs://{PROJECT_ID}-tmp-{now}\"\n",
+        "    BUCKET_URI = f\"gs://{PROJECT_ID}-tmp-{now}-{str(uuid.uuid4())[:4]}\"\n",
         "    ! gsutil mb -l {REGION} {BUCKET_URI}\n",
         "else:\n",
         "    BUCKET_NAME = \"/\".join(BUCKET_URI.split(\"/\")[:3])\n",