Update TPU CI container image (#9434)

bhavya01 · web-flow · commit 1c00dea2c9ad · 2025-07-02T11:24:41.000-07:00
diff --git a/infra/tpu-pytorch/tpu_ci.tf b/infra/tpu-pytorch/tpu_ci.tf
@@ -9,5 +9,6 @@ module "v4_arc_cluster" {
   max_tpu_nodes     = 32
   github_repo_url   = "https://github.com/pytorch/xla"
   # Dockerfile for this image can be found at test/tpu/Dockerfile
-  runner_image      = "gcr.io/tpu-pytorch/tpu-ci-runner:latest"
+  # TODO(https://github.com/pytorch/xla/issues/8946): Update latest tag
+  runner_image      = "gcr.io/tpu-pytorch/tpu-ci-runner:test"
 }

Original file line number	Diff line number	Diff line change
`@@ -9,5 +9,6 @@ module "v4_arc_cluster" {`
`9`	`9`	`max_tpu_nodes = 32`
`10`	`10`	`github_repo_url = "https://github.com/pytorch/xla"`
`11`	`11`	`# Dockerfile for this image can be found at test/tpu/Dockerfile`
`12`		`- runner_image = "gcr.io/tpu-pytorch/tpu-ci-runner:latest"`
	`12`	`+ # TODO(https://github.com/pytorch/xla/issues/8946): Update latest tag`
	`13`	`+ runner_image = "gcr.io/tpu-pytorch/tpu-ci-runner:test"`
`13`	`14`	`}`