From 71be46e7d8df84bc567aa177d6b41759e2bd1870 Mon Sep 17 00:00:00 2001
From: gmaninatarajan <138732032+gmaninatarajan@users.noreply.github.com>
Date: Thu, 30 Apr 2026 07:28:04 -0500
Subject: [PATCH] feat: Updated whl file and package name as part of Vertex
 Model Garden setup (#4508)

---
 .../cloudai_alphagenome_vai_quickstart.ipynb  | 37 ++++++-------------
 1 file changed, 12 insertions(+), 25 deletions(-)

diff --git a/notebooks/community/alphagenome/cloudai_alphagenome_vai_quickstart.ipynb b/notebooks/community/alphagenome/cloudai_alphagenome_vai_quickstart.ipynb
index 770de5359..7e045edbf 100644
--- a/notebooks/community/alphagenome/cloudai_alphagenome_vai_quickstart.ipynb
+++ b/notebooks/community/alphagenome/cloudai_alphagenome_vai_quickstart.ipynb
@@ -107,9 +107,8 @@
         "gcp_project = \"\"  # @param {type:\"string\"}\n",
         "vertex_ai_url = \"\"  # @param {type:\"string\"}\n",
         "\n",
-        "\n",
-        "code_whl = \"alphagenome-0.4.2.8-py3-none-any.whl\"\n",
-        "file_path = f\"gs://alphagenome-whl/{code_whl}\"\n"
+        "code_whl = \"alphagenome_cloud_sdk-0.4.2.9-py3-none-any.whl\"\n",
+        "file_path = f\"gs://alphagenome-whl/{code_whl}\""
       ]
     },
     {
@@ -173,8 +172,7 @@
         "\n",
         "# # Set the credential\n",
         "# !gcloud auth activate-service-account --key-file={service_account_key_file}\n",
-        "# os.environ[\"GOOGLE_APPLICATION_CREDENTIALS\"] = service_account_key_file\n",
-        "\n"
+        "# os.environ[\"GOOGLE_APPLICATION_CREDENTIALS\"] = service_account_key_file"
       ]
     },
     {
@@ -196,7 +194,7 @@
       "source": [
         "# Download and install Install AlphaGenome for Vertex AI\n",
         "print(f\"Downloading the wheel file: {file_path}\")\n",
-        "!gcloud storage cp {file_path} . --billing-project={gcp_project}\n"
+        "!gcloud storage cp {file_path} . --billing-project={gcp_project}"
       ]
     },
     {
@@ -210,7 +208,7 @@
         "# @markdown Run this cell to install AlphaGenome.\n",
         "from IPython.display import clear_output\n",
         "\n",
-        "! pip install $code_whl\n",
+        "! pip install $code_whl --force-reinstall\n",
         "clear_output()"
       ]
     },
@@ -237,9 +235,9 @@
         "from alphagenome.data import transcript as transcript_utils\n",
         "from alphagenome.data.genome import Interval\n",
         "from alphagenome.interpretation import ism\n",
-        "from alphagenome.models import (dna_client, dna_client_http, interval_scorers,\n",
-        "                                variant_scorers)\n",
-        "from alphagenome.visualization import plot_components"
+        "from alphagenome.models import dna_client, interval_scorers, variant_scorers\n",
+        "from alphagenome.visualization import plot_components\n",
+        "from alphagenome_cloud_sdk.models import dna_client_http"
       ]
     },
     {
@@ -1431,7 +1429,9 @@
       },
       "outputs": [],
       "source": [
-        "interval = Interval(chromosome=\"chr19\", start=2**20, end=2**20 + 2**20, strand=\".\")\n",
+        "interval = Interval(\n",
+        "    chromosome=\"chr19\", start=2**20, end=2**20 + 2**20, strand=\".\"\n",
+        ")\n",
         "\n",
         "scores = dna_model.score_interval(\n",
         "    interval,\n",
@@ -1585,21 +1585,8 @@
       "toc_visible": true
     },
     "kernelspec": {
-      "display_name": ".venv (3.13.12)",
-      "language": "python",
+      "display_name": "Python 3",
       "name": "python3"
-    },
-    "language_info": {
-      "codemirror_mode": {
-        "name": "ipython",
-        "version": 3
-      },
-      "file_extension": ".py",
-      "mimetype": "text/x-python",
-      "name": "python",
-      "nbconvert_exporter": "python",
-      "pygments_lexer": "ipython3",
-      "version": "3.13.12"
     }
   },
   "nbformat": 4,