Run ansible release stage after local builds are installed (#7955)

infra/ansible/e2e_tests.Dockerfile

@@ -13,13 +13,6 @@ RUN ansible-playbook -vvv playbook.yaml -e "stage=build" -e "${ansible_vars}"
 RUN ansible-playbook -vvv playbook.yaml -e "stage=build_plugin" -e "${ansible_vars}" --skip-tags=fetch_srcs,install_deps
 
 FROM python:${python_version}-${debian_version}
-WORKDIR /ansible
-RUN pip install ansible
-COPY . /ansible
-
-# Install runtime pip and apt dependencies.
-ARG ansible_vars
-RUN ansible-playbook -vvv playbook.yaml -e "stage=release" -e "${ansible_vars}" --tags "install_deps"
 
 # Copy test sources.
 RUN mkdir -p /src/pytorch/xla
@@ -33,6 +26,13 @@ COPY --from=build /dist/*.whl ./
 RUN echo "Installing the following wheels" && ls *.whl
 RUN pip install *.whl
 
+WORKDIR /ansible
+RUN pip install ansible
+COPY . /ansible
+# Install runtime pip and apt dependencies.
+ARG ansible_vars
+RUN ansible-playbook -vvv playbook.yaml -e "stage=release" -e "${ansible_vars}" --tags "install_deps"
+
 WORKDIR /
 
 # Clean-up unused directories.

infra/tpu-pytorch/test_triggers.tf

@@ -33,6 +33,7 @@ module "tpu_e2e_tests" {
     # The commit ID associated with the triggered build. Substituted when
     # Cloud Build is triggered.
     xla_git_rev = "$COMMIT_SHA"
+    bundle_libtpu = "0"
   }
 
   # Substitutions used in the "run_e2e_tests" step, see

setup.py

@@ -292,6 +292,7 @@ def run(self):
         'absl-py>=1.0.0',
         'numpy',
         'pyyaml',
+        'requests',
         # importlib.metadata backport required for PJRT plugin discovery prior
         # to Python 3.10
         'importlib_metadata>=4.6;python_version<"3.10"',