From bac5fda4bc7eafa9150c4a29fd81c530edda4125 Mon Sep 17 00:00:00 2001
From: Adrien <adrien@huggingface.co>
Date: Thu, 18 Jul 2024 18:05:27 +0200
Subject: [PATCH] wip

Signed-off-by: Adrien <adrien@huggingface.co>
---
 .github/workflows/build.yaml | 1 +
 1 file changed, 1 insertion(+)

diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml
index a633812d..8fb4c1e9 100644
--- a/.github/workflows/build.yaml
+++ b/.github/workflows/build.yaml
@@ -210,6 +210,7 @@ jobs:
           export DOCKER_IMAGE=registry-us-east-1.prod.aws.ci.huggingface.tech/api-inference/community/text-generation-inference:sha-c55c43e
           export HF_TOKEN=${{ secrets.HF_TOKEN }}
           echo $DOCKER_IMAGE
+          docker pull $DOCKER_IMAGE
           pytest -x -s -vv integration-tests ${PYTEST_FLAGS}
       - name: Setup tmate session
         if: ${{ failure() }}