In the referenced location the example uses the wrong method. CustomTrainingJob is used instead of CustomContainerTrainingJob. https://github.com/googleapis/python-aiplatform/blob/a9a8f44144749d33da585719a87b7208c60303b5/google/cloud/aiplatform/training_jobs.py#L2409