From 42ce781f908c96f396005ad242a04ba1c685e165 Mon Sep 17 00:00:00 2001
From: OlivierDehaene <23298448+OlivierDehaene@users.noreply.github.com>
Date: Mon, 15 May 2023 17:42:22 +0200
Subject: [PATCH] fix workflow

---
 .github/workflows/build.yaml | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml
index 19dc371d..c50ccf8c 100644
--- a/.github/workflows/build.yaml
+++ b/.github/workflows/build.yaml
@@ -191,8 +191,6 @@ jobs:
       - start-runner
       - build-and-push-image # Wait for the docker image to be built
     runs-on: ${{ needs.start-runner.outputs.label }} # run the job on the newly created runner
-    env:
-      DOCKER_IMAGE: registry.internal.huggingface.tech/api-inference/community/text-generation-inference:sha-${{ env.GITHUB_SHA_SHORT }}
     steps:
       - uses: actions/checkout@v2
       - name: Set up Python
@@ -214,6 +212,7 @@ jobs:
           make install-integration-tests
       - name: Run tests
         run: |
+          export DOCKER_IMAGE=registry.internal.huggingface.tech/api-inference/community/text-generation-inference:sha-${{ env.GITHUB_SHA_SHORT }}
           make integration-tests
 
   stop-runner: