docs: fix CustomContainerTrainingJob example in docstring (#1101)

Fixes #1100 🦕
googleapis · Mar 25, 2022 · d2fb9db · d2fb9db
1 parent ef5930c
commit d2fb9db
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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',