Skip to content

Commit 445df82

Browse files
authored
chore: Update Model Garden SDK to GA version (#13517)
* chore: Update Model Garden SDK to GA version * update required package version
1 parent 5749bac commit 445df82

File tree

5 files changed

+5
-5
lines changed

5 files changed

+5
-5
lines changed

model_garden/gemma/gemma3_deploy.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ def deploy() -> aiplatform.Endpoint:
2626
# [START aiplatform_modelgarden_gemma3_deploy]
2727

2828
import vertexai
29-
from vertexai.preview import model_garden
29+
from vertexai import model_garden
3030

3131
# TODO(developer): Update and un-comment below lines
3232
# PROJECT_ID = "your-project-id"

model_garden/gemma/models_deploy_options_list.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ def list_deploy_options(model : str) -> List[types.PublisherModel.CallToAction.D
2828
# [START aiplatform_modelgarden_models_deployables_options_list]
2929

3030
import vertexai
31-
from vertexai.preview import model_garden
31+
from vertexai import model_garden
3232

3333
# TODO(developer): Update and un-comment below lines
3434
# PROJECT_ID = "your-project-id"

model_garden/gemma/models_deployable_list.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ def list_deployable_models() -> List[str]:
2626
# [START aiplatform_modelgarden_models_deployables_list]
2727

2828
import vertexai
29-
from vertexai.preview import model_garden
29+
from vertexai import model_garden
3030

3131
# TODO(developer): Update and un-comment below lines
3232
# PROJECT_ID = "your-project-id"

model_garden/gemma/requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
google-cloud-aiplatform[all]==1.84.0
1+
google-cloud-aiplatform[all]==1.103.0

model_garden/gemma/test_model_garden_examples.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ def test_list_deploy_options() -> None:
3434
assert len(deploy_options) > 0
3535

3636

37-
@patch("vertexai.preview.model_garden.OpenModel")
37+
@patch("vertexai.model_garden.OpenModel")
3838
def test_gemma3_deploy(mock_open_model: MagicMock) -> None:
3939
# Mock the deploy response.
4040
mock_endpoint = aiplatform.Endpoint(endpoint_name="test-endpoint-name")

0 commit comments

Comments
 (0)