-
Notifications
You must be signed in to change notification settings - Fork 30
120 lines (97 loc) · 2.88 KB
/
test.yml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
name: "Test"
on:
workflow_dispatch:
pull_request:
branches:
- main
paths-ignore:
- "**.md"
- "proto/**"
defaults:
run:
shell: bash
env:
CI: true
DOCKER_BUILDKIT: 1
TEST_IMAGE_NAME: "cpu-tests:0"
jobs:
build:
runs-on: ubuntu-latest
steps:
- name: "Checkout"
uses: actions/checkout@v4
- name: "Free up disk space"
uses: ./.github/actions/free-up-disk-space
- name: "Set up QEMU"
uses: docker/setup-qemu-action@v3
- name: "Setup Docker Buildx"
uses: docker/setup-buildx-action@v3
- name: "Build test image"
uses: docker/build-push-action@v5
with:
context: .
file: ./Dockerfile
target: "cpu-tests"
tags: ${{ env.TEST_IMAGE_NAME }}
outputs: type=docker,dest=/tmp/test_image.tar
- name: "Upload test image"
uses: actions/upload-artifact@v4
with:
name: "test-image"
path: /tmp/test_image.tar
retention-days: 1
test-python:
runs-on: ubuntu-latest
needs: build
steps:
- name: "Checkout"
uses: actions/checkout@v3
- name: "Free up disk space"
uses: ./.github/actions/free-up-disk-space
- name: "Setup Docker Buildx"
uses: docker/setup-buildx-action@v3
- name: "Download test image"
uses: actions/download-artifact@v3
with:
name: "test-image"
path: /tmp
- name: "Load Docker image"
run: |
docker load --input /tmp/test_image.tar
docker image ls -a
- name: "Run Python tests"
run: |
#make python-tests
docker run --rm \
-v /tmp/transformers_cache:/transformers_cache \
-e HUGGINGFACE_HUB_CACHE=/transformers_cache \
-e TRANSFORMERS_CACHE=/transformers_cache \
$TEST_IMAGE_NAME pytest -sv \
--ignore=server/tests/test_utils.py \
--ignore=server/tests/models/test_bloom.py \
--ignore=server/tests/models/test_causal_lm.py \
--ignore=server/tests/models/test_seq2seq_lm.py \
--ignore=server/tests/utils/test_convert.py \
-k "not cache_with_eviction and not tensor_too_large" \
server/tests
integration-tests:
runs-on: ubuntu-latest
needs: build
steps:
- name: "Checkout"
uses: actions/checkout@v3
- name: "Free up disk space"
uses: ./.github/actions/free-up-disk-space
- name: "Setup Docker Buildx"
uses: docker/setup-buildx-action@v3
- name: "Download test image"
uses: actions/download-artifact@v3
with:
name: "test-image"
path: /tmp
- name: "Load Docker image"
run: |
docker load --input /tmp/test_image.tar
docker image ls -a
- name: "Run integration tests"
run: make integration-tests