Signed-off-by: Adrien <adrien@huggingface.co>
This commit is contained in:
Adrien 2024-07-17 22:21:10 +02:00
parent 7d081e1b36
commit 65dc1b9fbe
No known key found for this signature in database
GPG Key ID: 13A249A288DE0DAB

View File

@ -154,13 +154,8 @@ 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: ${{ needs.build-and-push.outputs.runs_on }}
if: needs.build-and-push.outputs.runs_on != 'ubuntu-latest'
group: aws-use1-g5-12xlarge
env:
PYTEST_FLAGS: ${{ (startsWith(github.ref, 'refs/tags/') || github.ref == 'refs/heads/main' || inputs.release-tests == true) && '--release' || '' }}
steps:
@ -178,8 +173,7 @@ jobs:
- name: Run tests
run: |
export DOCKER_VOLUME=/mnt/cache
export DOCKER_IMAGE=${{ needs.build-and-push.outputs.docker_image }}
export DOCKER_DEVICES=${{ needs.build-and-push.outputs.docker_devices }}
export DOCKER_IMAGE=registry-push.github-runners.huggingface.tech/api-inference/community/text-generation-inference:sha-c3e7bcc
export HF_TOKEN=${{ secrets.HF_TOKEN }}
echo $DOCKER_IMAGE
pytest -x -s -vv integration-tests ${PYTEST_FLAGS}