diff --git a/google/cloud/aiplatform/constants/prediction.py b/google/cloud/aiplatform/constants/prediction.py index 62f4a3e8cb..88ae2fd5ed 100644 --- a/google/cloud/aiplatform/constants/prediction.py +++ b/google/cloud/aiplatform/constants/prediction.py @@ -70,6 +70,9 @@ ] SKLEARN_CONTAINER_URIS = [ + "us-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-6:latest", + "europe-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-6:latest", + "asia-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-6:latest", "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",