Skip to content

Commit d2fb9db

Browse files
authored
docs: fix CustomContainerTrainingJob example in docstring (#1101)
Fixes #1100 🦕
1 parent ef5930c commit d2fb9db

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

google/cloud/aiplatform/training_jobs.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -2406,9 +2406,9 @@ def __init__(
24062406
):
24072407
"""Constructs a Custom Container Training Job.
24082408
2409-
job = aiplatform.CustomTrainingJob(
2409+
job = aiplatform.CustomContainerTrainingJob(
24102410
display_name='test-train',
2411-
container_uri='gcr.io/cloud-aiplatform/training/tf-cpu.2-2:latest',
2411+
container_uri='gcr.io/my_project_id/my_image_name:tag',
24122412
command=['python3', 'run_script.py']
24132413
model_serving_container_image_uri='gcr.io/my-trainer/serving:1',
24142414
model_serving_container_predict_route='predict',

0 commit comments

Comments
 (0)