diff --git a/infra/tpu-pytorch/tpu_ci.tf b/infra/tpu-pytorch/tpu_ci.tf index 4c724660c0ee..fab86b18f60d 100644 --- 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" }