From aa0b19ea8d931198081061db5db8e7919867e54e Mon Sep 17 00:00:00 2001 From: Adrien Date: Wed, 17 Jul 2024 22:33:04 +0200 Subject: [PATCH] wip Signed-off-by: Adrien --- .github/workflows/build.yaml | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml index 459e9b80..a85e86e3 100644 --- a/.github/workflows/build.yaml +++ b/.github/workflows/build.yaml @@ -154,8 +154,13 @@ jobs: echo "runs_on=${{ env.RUNS_ON }}" >> "$GITHUB_OUTPUT" echo "label=${{ env.LABEL }}" >> "$GITHUB_OUTPUT" integration_tests: + concurrency: + group: ${{ github.workflow }}-${{ github.job }}-${{ needs.build-and-push.outputs.label }}-${{ github.head_ref || github.run_id }} + cancel-in-progress: true + needs: build-and-push runs-on: - group: aws-use1-g5-12xlarge + group: ${{ needs.build-and-push.outputs.runs_on }} + if: needs.build-and-push.outputs.runs_on != 'ubuntu-latest' env: PYTEST_FLAGS: ${{ (startsWith(github.ref, 'refs/tags/') || github.ref == 'refs/heads/main' || inputs.release-tests == true) && '--release' || '' }} steps: @@ -173,7 +178,8 @@ jobs: - name: Run tests run: | export DOCKER_VOLUME=/mnt/cache - export DOCKER_IMAGE=registry-push.github-runners.huggingface.tech/api-inference/community/text-generation-inference:sha-c3e7bcc + export DOCKER_IMAGE=${{ needs.build-and-push.outputs.docker_image }} + export DOCKER_DEVICES=${{ needs.build-and-push.outputs.docker_devices }} export HF_TOKEN=${{ secrets.HF_TOKEN }} echo $DOCKER_IMAGE - pytest -x -s -vv integration-tests ${PYTEST_FLAGS} + pytest -s -vv integration-tests ${PYTEST_FLAGS}