From 1efc0dd6d380de38a154a7f6f989c93c6eb45777 Mon Sep 17 00:00:00 2001 From: Morgan Du Date: Tue, 22 Mar 2022 10:56:16 -0700 Subject: [PATCH] fix: CustomContainerTrainingJob example in docstring --- google/cloud/aiplatform/training_jobs.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/google/cloud/aiplatform/training_jobs.py b/google/cloud/aiplatform/training_jobs.py index f84a4136d8..fa1e276ddd 100644 --- a/google/cloud/aiplatform/training_jobs.py +++ b/google/cloud/aiplatform/training_jobs.py @@ -2406,9 +2406,9 @@ def __init__( ): """Constructs a Custom Container Training Job. - job = aiplatform.CustomTrainingJob( + job = aiplatform.CustomContainerTrainingJob( display_name='test-train', - container_uri='gcr.io/cloud-aiplatform/training/tf-cpu.2-2:latest', + container_uri='gcr.io/my_project_id/my_image_name:tag', command=['python3', 'run_script.py'] model_serving_container_image_uri='gcr.io/my-trainer/serving:1', model_serving_container_predict_route='predict',