Skip to content

Migrate gsutil usage to gcloud storage #4314

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
gericdong merged 9 commits into from
Dec 24, 2025
Merged

Migrate gsutil usage to gcloud storage #4314

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
9 commits
Select commit Hold shift + click to select a range
7e165da
Migrate gsutil usage to gcloud storage
googlyrahman Oct 16, 2025
7e2db68
changes for 4314
gurusai-voleti Dec 12, 2025
8589909
Apply automated linter fixes
gurusai-voleti Dec 16, 2025
50db888
remove unused imports
gurusai-voleti Dec 16, 2025
427dcfc
Merge pull request #49 from gurusai-voleti/br_4314
googlyrahman Dec 17, 2025
bb31a00
removed model_garden changes
gurusai-voleti Dec 22, 2025
38e3c2c
Merge pull request #84 from gurusai-voleti/br_4314
gurusai-voleti Dec 22, 2025
cff974d
updates
gurusai-voleti Dec 22, 2025
69bf480
Merge pull request #85 from gurusai-voleti/br_4314
gurusai-voleti Dec 22, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
22 changes: 7 additions & 15 deletions notebooks/community/migration/UJ15 legacy AutoML Vision Video Object Tracking.ipynb
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -340,8 +340,7 @@
},
"outputs": [],
"source": [
"! gsutil mb -l $REGION gs://$BUCKET_NAME\n"
]
"! gcloud storage buckets create --location $REGION gs://$BUCKET_NAME\n" ]
},
{
"cell_type": "markdown",
Expand All @@ -361,8 +360,7 @@
},
"outputs": [],
"source": [
"! gsutil ls -al gs://$BUCKET_NAME\n"
]
"! gcloud storage ls --all-versions --long gs://$BUCKET_NAME\n" ]
},
{
"cell_type": "markdown",
Expand Down Expand Up @@ -506,8 +504,7 @@
},
"outputs": [],
"source": [
"! gsutil cat $IMPORT_FILE | head -n 10\n"
]
"! gcloud storage cat $IMPORT_FILE | head -n 10\n" ]
},
{
"cell_type": "markdown",
Expand Down Expand Up @@ -1222,8 +1219,7 @@
"source": [
"TRAIN_FILES = \"gs://automl-video-demo-data/traffic_videos/traffic_videos_labels.csv\"\n",
"\n",
"test_items = ! gsutil cat $TRAIN_FILES | head -n2\n",
"\n",
"test_items = ! gcloud storage cat $TRAIN_FILES | head -n2\n", "\n",
"cols = str(test_items[0]).split(',')\n",
"test_item_1 = str(cols[0])\n",
"test_label_1 = str(cols[1])\n",
Expand Down Expand Up @@ -1286,8 +1282,7 @@
" f.write(data + '\\n')\n",
" \n",
"print(gcs_input_uri)\n",
"!gsutil cat $gcs_input_uri\n"
]
"!gcloud storage cat $gcs_input_uri\n" ]
},
{
"cell_type": "markdown",
Expand Down Expand Up @@ -1452,9 +1447,7 @@
"source": [
"destination_uri = batch_prediction.output_config.gcs_destination.output_uri_prefix[:-1]\n",
"\n",
"! gsutil ls $destination_uri/prediction-**\n",
"! gsutil cat $destination_uri/prediction-**"
]
"! gcloud storage ls $destination_uri/prediction-**\n", "! gcloud storage cat $destination_uri/prediction-**" ]
},
{
"cell_type": "markdown",
Expand Down Expand Up @@ -1548,8 +1541,7 @@
" print(e)\n",
"\n",
"if delete_bucket and 'BUCKET_NAME' in globals():\n",
" ! gsutil rm -r gs://$BUCKET_NAME\n"
]
" ! gcloud storage rm --recursive gs://$BUCKET_NAME\n" ]
},
{
"cell_type": "code",
Expand Down
25 changes: 8 additions & 17 deletions notebooks/community/ml_ops/stage6/get_started_with_automl_image_model_batch.ipynb
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -483,8 +483,7 @@
},
"outputs": [],
"source": [
"! gsutil mb -l $REGION $BUCKET_URI"
]
"! gcloud storage buckets create --location $REGION $BUCKET_URI" ]
},
{
"cell_type": "markdown",
Expand All @@ -503,8 +502,7 @@
},
"outputs": [],
"source": [
"! gsutil ls -al $BUCKET_URI"
]
"! gcloud storage ls --all-versions --long $BUCKET_URI" ]
},
{
"cell_type": "markdown",
Expand Down Expand Up @@ -612,12 +610,10 @@
"else:\n",
" FILE = IMPORT_FILE\n",
"\n",
"count = ! gsutil cat $FILE | wc -l\n",
"print(\"Number of Examples\", int(count[0]))\n",
"count = ! gcloud storage cat $FILE | wc -l\n", "print(\"Number of Examples\", int(count[0]))\n",
"\n",
"print(\"First 10 rows\")\n",
"! gsutil cat $FILE | head"
]
"! gcloud storage cat $FILE | head" ]
},
{
"cell_type": "markdown",
Expand Down Expand Up @@ -869,8 +865,7 @@
},
"outputs": [],
"source": [
"test_items = !gsutil cat $IMPORT_FILE | head -n2\n",
"if len(str(test_items[0]).split(\",\")) == 3:\n",
"test_items = !gcloud storage cat $IMPORT_FILE | head -n2\n", "if len(str(test_items[0]).split(\",\")) == 3:\n",
" _, test_item_1, test_label_1 = str(test_items[0]).split(\",\")\n",
" _, test_item_2, test_label_2 = str(test_items[1]).split(\",\")\n",
"else:\n",
Expand Down Expand Up @@ -903,9 +898,7 @@
"file_1 = test_item_1.split(\"/\")[-1]\n",
"file_2 = test_item_2.split(\"/\")[-1]\n",
"\n",
"! gsutil cp $test_item_1 $BUCKET_URI/$file_1\n",
"! gsutil cp $test_item_2 $BUCKET_URI/$file_2\n",
"\n",
"! gcloud storage cp $test_item_1 $BUCKET_URI/$file_1\n", "! gcloud storage cp $test_item_2 $BUCKET_URI/$file_2\n", "\n",
"test_item_1 = BUCKET_URI + \"/\" + file_1\n",
"test_item_2 = BUCKET_URI + \"/\" + file_2"
]
Expand Down Expand Up @@ -948,8 +941,7 @@
" f.write(json.dumps(data) + \"\\n\")\n",
"\n",
"print(gcs_input_uri)\n",
"! gsutil cat $gcs_input_uri"
]
"! gcloud storage cat $gcs_input_uri" ]
},
{
"cell_type": "markdown",
Expand Down Expand Up @@ -1106,8 +1098,7 @@
"delete_bucket = False\n",
"\n",
"if delete_bucket or os.getenv(\"IS_TESTING\"):\n",
" ! gsutil rm -rf {BUCKET_URI}"
]
" ! gcloud storage rm --recursive --continue-on-error {BUCKET_URI}" ]
}
],
"metadata": {
Expand Down
18 changes: 6 additions & 12 deletions notebooks/community/ml_ops/stage6/get_started_with_automl_image_model_online.ipynb
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -479,8 +479,7 @@
},
"outputs": [],
"source": [
"! gsutil mb -l $REGION $BUCKET_URI"
]
"! gcloud storage buckets create --location $REGION $BUCKET_URI" ]
},
{
"cell_type": "markdown",
Expand All @@ -499,8 +498,7 @@
},
"outputs": [],
"source": [
"! gsutil ls -al $BUCKET_URI"
]
"! gcloud storage ls --all-versions --long $BUCKET_URI" ]
},
{
"cell_type": "markdown",
Expand Down Expand Up @@ -685,12 +683,10 @@
"else:\n",
" FILE = IMPORT_FILE\n",
"\n",
"count = ! gsutil cat $FILE | wc -l\n",
"print(\"Number of Examples\", int(count[0]))\n",
"count = ! gcloud storage cat $FILE | wc -l\n", "print(\"Number of Examples\", int(count[0]))\n",
"\n",
"print(\"First 10 rows\")\n",
"! gsutil cat $FILE | head"
]
"! gcloud storage cat $FILE | head" ]
},
{
"cell_type": "markdown",
Expand Down Expand Up @@ -969,8 +965,7 @@
},
"outputs": [],
"source": [
"test_item = !gsutil cat $IMPORT_FILE | head -n1\n",
"if len(str(test_item[0]).split(\",\")) == 3:\n",
"test_item = !gcloud storage cat $IMPORT_FILE | head -n1\n", "if len(str(test_item[0]).split(\",\")) == 3:\n",
" _, test_item, test_label = str(test_item[0]).split(\",\")\n",
"else:\n",
" test_item, test_label = str(test_item[0]).split(\",\")\n",
Expand Down Expand Up @@ -1122,8 +1117,7 @@
"delete_bucket = False\n",
"\n",
"if delete_bucket or os.getenv(\"IS_TESTING\"):\n",
" ! gsutil rm -rf {BUCKET_URI}"
]
" ! gcloud storage rm --recursive --continue-on-error {BUCKET_URI}" ]
}
],
"metadata": {
Expand Down
12 changes: 4 additions & 8 deletions notebooks/community/ml_ops/stage6/get_started_with_tf_serving_function.ipynb
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -443,8 +443,7 @@
},
"outputs": [],
"source": [
"! gsutil mb -l $REGION $BUCKET_URI"
]
"! gcloud storage buckets create --location=$REGION $BUCKET_URI" ]
Copy link
Contributor

@gemini-code-assist gemini-code-assist bot Oct 16, 2025

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

medium

For consistency with other commands in this pull request and general gcloud command-line style, it's better to use a space to separate the --location flag and its value, rather than an equals sign.

        "! gcloud storage buckets create --location $REGION $BUCKET_URI"

},
{
"cell_type": "markdown",
Expand All @@ -463,8 +462,7 @@
},
"outputs": [],
"source": [
"! gsutil ls -al $BUCKET_URI"
]
"! gcloud storage ls --all-versions --long $BUCKET_URI" ]
},
{
"cell_type": "markdown",
Expand Down Expand Up @@ -947,8 +945,7 @@
},
"outputs": [],
"source": [
"! gsutil cp gs://cloud-ml-data/img/flower_photos/daisy/100080576_f52e8ee070_n.jpg test.jpg"
]
"! gcloud storage cp gs://cloud-ml-data/img/flower_photos/daisy/100080576_f52e8ee070_n.jpg test.jpg" ]
},
{
"cell_type": "code",
Expand Down Expand Up @@ -1050,8 +1047,7 @@
" print(e)\n",
"\n",
"if delete_bucket or os.getenv(\"IS_TESTING\"):\n",
" ! gsutil rm -rf {BUCKET_URI}"
]
" ! gcloud storage rm --recursive --continue-on-error {BUCKET_URI}" ]
}
],
"metadata": {
Expand Down
18 changes: 8 additions & 10 deletions notebooks/community/model_garden/model_garden_codegemma_deployment_on_vertex.ipynb
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -154,11 +154,10 @@
"if BUCKET_URI is None or BUCKET_URI.strip() == \"\" or BUCKET_URI == \"gs://\":\n",
" BUCKET_URI = f\"gs://{PROJECT_ID}-tmp-{now}-{str(uuid.uuid4())[:4]}\"\n",
" BUCKET_NAME = \"/\".join(BUCKET_URI.split(\"/\")[:3])\n",
" ! gsutil mb -l {REGION} {BUCKET_URI}\n",
"else:\n",
" ! gcloud storage buckets create --location {REGION} {BUCKET_URI}\n", "else:\n",
" assert BUCKET_URI.startswith(\"gs://\"), \"BUCKET_URI must start with `gs://`.\"\n",
" shell_output = ! gsutil ls -Lb {BUCKET_NAME} | grep \"Location constraint:\" | sed \"s/Location constraint://\"\n",
" bucket_region = shell_output[0].strip().lower()\n",
# Note: The format of the full listing output is different. gcloud storage uses a title case for keys and will not display a field if its value is "None".
" shell_output = ! gcloud storage ls --full --buckets {BUCKET_NAME} | grep \"Location constraint:\" | sed \"s/Location constraint://\"\n", " bucket_region = shell_output[0].strip().lower()\n",
Copy link
Contributor

@gemini-code-assist gemini-code-assist bot Oct 16, 2025

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

high

This command will fail because it's parsing the output of gcloud storage ls incorrectly. The note on the previous line correctly points out that gcloud storage uses title case for keys, so grep should be looking for Location Constraint: instead of Location constraint:. The sed command also needs to be updated to match the title case string.

    shell_output = ! gcloud storage ls --full --buckets {BUCKET_NAME} | grep "Location Constraint:" | sed "s/Location Constraint: *//"

" if bucket_region != REGION:\n",
" raise ValueError(\n",
" \"Bucket region %s is different from notebook region %s\"\n",
Expand All @@ -182,8 +181,9 @@
"\n",
"\n",
"# Provision permissions to the SERVICE_ACCOUNT with the GCS bucket\n",
"! gsutil iam ch serviceAccount:{SERVICE_ACCOUNT}:roles/storage.admin $BUCKET_NAME\n",
"\n",
"! # Note: Migrating scripts using gsutil iam ch is more complex than get or set. You need to replace the single iam ch command with a series of gcloud storage bucket add-iam-policy-binding and/or gcloud storage bucket remove-iam-policy-binding commands, or replicate the read-modify-write loop.\n",
"! # Note: Conditions: gsutil iam ch does not support modifying IAM policies that contain conditions. gcloud storage commands do support conditions.\n",
"! gcloud storage buckets add-iam-policy-binding $BUCKET_NAME --member=serviceAccount:{SERVICE_ACCOUNT} --role=roles/storage.admin\n", "\n",
"! gcloud config set project $PROJECT_ID\n",
"! gcloud projects add-iam-policy-binding --no-user-output-enabled {PROJECT_ID} --member=serviceAccount:{SERVICE_ACCOUNT} --role=\"roles/storage.admin\"\n",
"! gcloud projects add-iam-policy-binding --no-user-output-enabled {PROJECT_ID} --member=serviceAccount:{SERVICE_ACCOUNT} --role=\"roles/aiplatform.user\""
Expand Down Expand Up @@ -238,8 +238,7 @@
"\n",
" ! mkdir -p ./codegemma\n",
" ! curl -X GET \"{signed_url}\" | tar -xzvf - -C ./codegemma/\n",
" ! gsutil -m cp -R ./codegemma/* {BUCKET_URI}\n",
"\n",
" ! gcloud storage cp --recursive ./codegemma/* {BUCKET_URI}\n", "\n",
" model_path_prefix = BUCKET_URI.strip(\"/\") + \"/codegemma\"\n",
"else:\n",
" model_path_prefix = \"google/\""
Expand Down Expand Up @@ -820,8 +819,7 @@
"\n",
"delete_bucket = False # @param {type:\"boolean\"}\n",
"if delete_bucket:\n",
" ! gsutil -m rm -r $BUCKET_NAME"
]
" ! gcloud storage rm --recursive $BUCKET_NAME" ]
}
],
"metadata": {
Expand Down
3 changes: 1 addition & 2 deletions notebooks/community/model_garden/model_garden_reasoning_engine_llama3_1.ipynb
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -1106,8 +1106,7 @@
"delete_reasoning_engine = False # @param {type:\"boolean\"}\n",
"\n",
"if delete_bucket:\n",
" ! gsutil -m rm -r $BUCKET_NAME\n",
"\n",
" ! gcloud storage rm --recursive $BUCKET_NAME\n", "\n",
"if delete_reasoning_engine:\n",
" remote_agent.delete()"
]
Expand Down
30 changes: 10 additions & 20 deletions notebooks/community/sdk/sdk_automl_tabular_binary_classification_online_export_cloud.ipynb
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -472,8 +472,7 @@
},
"outputs": [],
"source": [
"! gsutil mb -l $REGION $BUCKET_NAME"
]
"! gcloud storage buckets create --location=$REGION $BUCKET_NAME" ]
Copy link
Contributor

@gemini-code-assist gemini-code-assist bot Oct 16, 2025

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

medium

For consistency with other commands in this pull request and general gcloud command-line style, it's better to use a space to separate the --location flag and its value, rather than an equals sign.

        "! gcloud storage buckets create --location $REGION $BUCKET_NAME"

},
{
"cell_type": "markdown",
Expand All @@ -492,8 +491,7 @@
},
"outputs": [],
"source": [
"! gsutil ls -al $BUCKET_NAME"
]
"! gcloud storage ls --all-versions --long $BUCKET_NAME" ]
},
{
"cell_type": "markdown",
Expand Down Expand Up @@ -596,14 +594,11 @@
},
"outputs": [],
"source": [
"count = ! gsutil cat $IMPORT_FILE | wc -l\n",
"print(\"Number of Examples\", int(count[0]))\n",
"count = ! gcloud storage cat $IMPORT_FILE | wc -l\n", "print(\"Number of Examples\", int(count[0]))\n",
"\n",
"print(\"First 10 rows\")\n",
"! gsutil cat $IMPORT_FILE | head\n",
"\n",
"heading = ! gsutil cat $IMPORT_FILE | head -n1\n",
"label_column = str(heading).split(\",\")[-1].split(\"'\")[0]\n",
"! gcloud storage cat $IMPORT_FILE | head\n", "\n",
"heading = ! gcloud storage cat $IMPORT_FILE | head -n1\n", "label_column = str(heading).split(\",\")[-1].split(\"'\")[0]\n",
"print(\"Label Column Name\", label_column)\n",
"if label_column is None:\n",
" raise Exception(\"label column missing\")"
Expand Down Expand Up @@ -832,15 +827,12 @@
"source": [
"print(\"Model Package:\", model_package)\n",
"print(\"Contents:\")\n",
"! gsutil ls $model_package\n",
"\n",
"! gcloud storage ls $model_package\n", "\n",
"print(\"\\nTF Saved Model\")\n",
"path = model_package + \"/predict\"\n",
"files = ! gsutil ls $path\n",
"saved_dir = files[1]\n",
"files = ! gcloud storage ls $path\n", "saved_dir = files[1]\n",
"print(saved_dir)\n",
"! gsutil ls $saved_dir"
]
"! gcloud storage ls $saved_dir" ]
},
{
"cell_type": "markdown",
Expand All @@ -861,8 +853,7 @@
},
"outputs": [],
"source": [
"! gsutil cp -r $model_package ."
]
"! gcloud storage cp --recursive $model_package ." ]
},
{
"cell_type": "markdown",
Expand Down Expand Up @@ -1155,8 +1146,7 @@
" print(e)\n",
"\n",
" if \"BUCKET_NAME\" in globals():\n",
" ! gsutil rm -r $BUCKET_NAME"
]
" ! gcloud storage rm --recursive $BUCKET_NAME" ]
}
],
"metadata": {
Expand Down
Loading
Loading