forked from kserve/kserve
-
Notifications
You must be signed in to change notification settings - Fork 0
147 lines (125 loc) · 3.9 KB
/
sklearnserver-docker-publish.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
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
name: Sklearn Server Docker Publisher
on:
push:
branches:
- master
- sklearn-runtime-schedule-retry-on-failure
tags:
- v*
pull_request:
workflow_run:
workflows: ["Sklearn Server Docker Publisher"]
types:
- completed
schedule:
- cron: '*/15 * * * *' # Runs every 15 minutes
env:
IMAGE_NAME: sklearnserver
concurrency:
group: ${{ github.workflow }}-${{ github.ref }}
cancel-in-progress: true
jobs:
# Main Test job
test:
runs-on: ubuntu-latest
strategy:
matrix:
platform: [linux/amd64, linux/arm64/v8, linux/ppc64le]
steps:
- name: Checkout source
uses: actions/checkout@v4
- name: Setup QEMU
uses: docker/setup-qemu-action@v3
- name: Setup Docker Buildx
uses: docker/setup-buildx-action@v3
- name: Restore cache
uses: actions/cache@v4
with:
path: |
~/.cache/pypoetry/cache
~/.cache/pypoetry/artifacts
key: ${{ runner.os }}-poetry-${{ hashFiles('**/poetry.lock') }}
restore-keys: |
${{ runner.os }}-poetry-
- name: Run tests
uses: docker/build-push-action@v5
with:
platforms: ${{ matrix.platform }}
context: python
file: python/sklearn.Dockerfile
push: false
cache-from: type=gha,mode=max
cache-to: type=gha,mode=max
provenance: false
timeout-minutes: 120 # Set the time limit for each job
continue-on-error: true
retry:
runs-on: ubuntu-latest
if: ${{ failure() && github.event.workflow_run.conclusion == 'timed_out' }}
needs: test
steps:
- name: Retry failed job due to timeout
run: echo "Retrying job due to timeout failure..."
- name: Checkout source
uses: actions/checkout@v4
- name: Setup QEMU
uses: docker/setup-qemu-action@v3
- name: Setup Docker Buildx
uses: docker/setup-buildx-action@v3
- name: Restore cache
uses: actions/cache@v4
with:
path: |
~/.cache/pypoetry/cache
~/.cache/pypoetry/artifacts
key: ${{ runner.os }}-poetry-${{ hashFiles('**/poetry.lock') }}
restore-keys: |
${{ runner.os }}-poetry-
- name: Run tests again
uses: docker/build-push-action@v5
with:
platforms: ${{ matrix.platform }}
context: python
file: python/sklearn.Dockerfile
push: false
cache-from: type=gha,mode=max
cache-to: type=gha,mode=max
provenance: false
# Push image to GitHub Packages.
push:
needs: test
runs-on: ubuntu-latest
strategy:
matrix:
platform: [linux/amd64, linux/arm64/v8, linux/ppc64le]
if: github.event_name == 'push' && !cancelled()
steps:
- name: Checkout source
uses: actions/checkout@v4
- name: Setup QEMU
uses: docker/setup-qemu-action@v3
- name: Setup Docker Buildx
uses: docker/setup-buildx-action@v3
- name: Export version variable
run: |
IMAGE_ID=kserve/$IMAGE_NAME
IMAGE_ID=$(echo $IMAGE_ID | tr '[A-Z]' '[a-z]')
VERSION=$(echo "${{ github.ref }}" | sed -e 's,.*/\(.*\),\1,')
[ "$VERSION" == "master" ] && VERSION=latest
echo VERSION=$VERSION >> $GITHUB_ENV
echo IMAGE_ID=$IMAGE_ID >> $GITHUB_ENV
- name: Build and push
uses: docker/build-push-action@v5
with:
platforms: ${{ matrix.platform }}
context: python
file: python/sklearn.Dockerfile
push: true
tags: ${{ env.IMAGE_ID }}:${{ env.VERSION }}
cache-from: type=gha,mode=max
cache-to: type=gha,mode=max
provenance: false
- name: Clear cache after successful build
if: success()
run: |
rm -rf ~/.cache/pypoetry/cache ~/.cache/pypoetry/artifacts