chore: fix CustomJob system test

PiperOrigin-RevId: 578020823

Author: jaycee-li

tests/system/aiplatform/test_custom_job.py

@@ -26,7 +26,9 @@
 from google.cloud.aiplatform.compat.types import job_state as gca_job_state
 from tests.system.aiplatform import e2e_base
 
-_PREBUILT_CONTAINER_IMAGE = "gcr.io/deeplearning-platform-release/base-cpu"
+_PREBUILT_CONTAINER_IMAGE = (
+    "us-docker.pkg.dev/vertex-ai/training/sklearn-cpu.1-0:latest"
+)
 _CUSTOM_CONTAINER_IMAGE = "python:3.8"
 
 _DIR_NAME = os.path.dirname(os.path.abspath(__file__))
@@ -35,8 +37,6 @@
 )
 
 
-# runtime patch the aiplatform path installed in CustomJob
-# remove the patch after aiplatform 1.24.0 is released
 @mock.patch.object(
     constants,
     "AIPLATFORM_DEPENDENCY_PATH",