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