From 9923f721bcfcb8b7258120d1f740778d5b9f435d Mon Sep 17 00:00:00 2001 From: Eric Dong Date: Thu, 31 Oct 2024 13:28:21 -0400 Subject: [PATCH] refactor: Update the batch prediction API namespace (#1365) # Description Update the batch prediction API namespace. The new API is available in Vertex AI SDK for Python 1.71.0. --- gemini/batch-prediction/intro_batch_prediction.ipynb | 2 +- .../intro_batch_prediction_using_bigquery_input.ipynb | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/gemini/batch-prediction/intro_batch_prediction.ipynb b/gemini/batch-prediction/intro_batch_prediction.ipynb index dbfe70d1b0..84bd972d81 100644 --- a/gemini/batch-prediction/intro_batch_prediction.ipynb +++ b/gemini/batch-prediction/intro_batch_prediction.ipynb @@ -253,7 +253,7 @@ "\n", "from google.cloud import storage\n", "from vertexai.generative_models import GenerativeModel\n", - "from vertexai.preview.batch_prediction import BatchPredictionJob" + "from vertexai.batch_prediction import BatchPredictionJob" ] }, { diff --git a/gemini/batch-prediction/intro_batch_prediction_using_bigquery_input.ipynb b/gemini/batch-prediction/intro_batch_prediction_using_bigquery_input.ipynb index 8f082edbe3..e83d094599 100644 --- a/gemini/batch-prediction/intro_batch_prediction_using_bigquery_input.ipynb +++ b/gemini/batch-prediction/intro_batch_prediction_using_bigquery_input.ipynb @@ -247,8 +247,8 @@ "\n", "from google.cloud import bigquery\n", "import vertexai\n", - "from vertexai.generative_models import GenerativeModel\n", - "from vertexai.preview.batch_prediction import BatchPredictionJob" + "from vertexai.batch_prediction import BatchPredictionJob\n", + "from vertexai.generative_models import GenerativeModel" ] }, {