Enable install google-cloud-sdk-gke-gcloud-auth-plugin to enable TP…

…UCI run_e2e_tests (#6420)
pytorch · Feb 1, 2024 · 479559c · 479559c
1 parent d686d65
commit 479559c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/infra/terraform_modules/xla_docker_build/xla_docker_build.tf b/infra/terraform_modules/xla_docker_build/xla_docker_build.tf
@@ -131,7 +131,8 @@ locals {
 
           # Try to setup kubectl credentials the new way,
           # see https://cloud.google.com/blog/products/containers-kubernetes/kubectl-auth-changes-in-gke
-          apt-get install google-cloud-sdk-gke-gcloud-auth-plugin -o DPkg::options::="--force-overwrite"
+          apt-get remove google-cloud-cli-gke-gcloud-auth-plugin
+          apt-get install google-cloud-sdk-gke-gcloud-auth-plugin -y -o DPkg::options::="--force-overwrite"
           export USE_GKE_GCLOUD_AUTH_PLUGIN=True
           apt-get install kubectl
           gcloud container clusters get-credentials $_CLUSTER_NAME --zone $_CLUSTER_ZONE