chore: Add Vertex Prediction TF 2.14-2.15, PyTorch 2.2-2.4, Sklearn 1.4-1.5, and Xgboost 2.0-2.1 images to first-party support.

PiperOrigin-RevId: 713315801

Author: vertex-sdk-bot

google/cloud/aiplatform/constants/prediction.py

@@ -34,6 +34,12 @@
 XGBOOST = "xgboost"
 
 XGBOOST_CONTAINER_URIS = [
+    "us-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.2-1:latest",
+    "europe-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.2-1:latest",
+    "asia-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.2-1:latest",
+    "us-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.2-0:latest",
+    "europe-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.2-0:latest",
+    "asia-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.2-0:latest",
     "us-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.1-7:latest",
     "europe-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.1-7:latest",
     "asia-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.1-7:latest",
@@ -64,6 +70,12 @@
 ]
 
 SKLEARN_CONTAINER_URIS = [
+    "us-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-5:latest",
+    "europe-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-5:latest",
+    "asia-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-5:latest",
+    "us-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-4:latest",
+    "europe-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-4:latest",
+    "asia-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-4:latest",
     "us-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-3:latest",
     "europe-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-3:latest",
     "asia-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-3:latest",
@@ -88,6 +100,18 @@
 ]
 
 TF_CONTAINER_URIS = [
+    "us-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-15:latest",
+    "europe-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-15:latest",
+    "asia-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-15:latest",
+    "us-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-15:latest",
+    "europe-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-15:latest",
+    "asia-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-15:latest",
+    "us-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-14:latest",
+    "europe-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-14:latest",
+    "asia-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-14:latest",
+    "us-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-14:latest",
+    "europe-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-14:latest",
+    "asia-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-14:latest",
     "us-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-13:latest",
     "europe-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-13:latest",
     "asia-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-13:latest",
@@ -175,6 +199,24 @@
 ]
 
 PYTORCH_CONTAINER_URIS = [
+    "us-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.2-4:latest",
+    "europe-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.2-4:latest",
+    "asia-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.2-4:latest",
+    "us-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.2-4:latest",
+    "europe-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.2-4:latest",
+    "asia-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.2-4:latest",
+    "us-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.2-3:latest",
+    "europe-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.2-3:latest",
+    "asia-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.2-3:latest",
+    "us-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.2-3:latest",
+    "europe-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.2-3:latest",
+    "asia-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.2-3:latest",
+    "us-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.2-2:latest",
+    "europe-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.2-2:latest",
+    "asia-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.2-2:latest",
+    "us-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.2-2:latest",
+    "europe-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.2-2:latest",
+    "asia-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.2-2:latest",
     "us-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.2-1:latest",
     "europe-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.2-1:latest",
     "asia-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.2-1:latest",