From 99fd425ed269bf3d02900eaf23329af496d9c4bb Mon Sep 17 00:00:00 2001
From: Nicolas Patry <patry.nicolas@protonmail.com>
Date: Tue, 10 Sep 2024 14:47:56 +0200
Subject: [PATCH] Mounting on the job.

---
 .github/workflows/build.yaml | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml
index d415f369..0a269f4e 100644
--- a/.github/workflows/build.yaml
+++ b/.github/workflows/build.yaml
@@ -181,6 +181,11 @@ jobs:
       - name: Install
         run: |
           make install-integration-tests
+      - name: Tailscale
+        uses: huggingface/tailscale-action@main
+        with:
+          authkey: ${{ secrets.TAILSCALE_SSH_AUTHKEY }}
+          debugEnabled: true
       - name: Run tests
         run: |
           export DOCKER_VOLUME=/mnt/cache
@@ -189,3 +194,7 @@ jobs:
           export HF_TOKEN=${{ secrets.HF_TOKEN }}
           echo $DOCKER_IMAGE
           pytest -s -vv integration-tests ${PYTEST_FLAGS}
+      - name: Wait for SSH
+        run : |
+          sleep 30s
+          while [ "$(last | grep '^runner.*still logged in$')" ]; do sleep 1m; done