File size: 1,841 Bytes
1078d3e
658ed86
1078d3e
dd83a20
1078d3e
 
5d931cc
 
f4868d7
dd83a20
5d931cc
 
f4868d7
1078d3e
d199d6c
 
1078d3e
 
 
 
 
 
 
 
 
 
 
 
 
d199d6c
658ed86
d35278a
658ed86
 
d199d6c
 
 
 
2b8c28b
d199d6c
 
2b8c28b
d199d6c
658ed86
72b6ca0
a653392
d199d6c
 
b6ab8aa
34c0a86
d199d6c
 
 
9218ebe
1078d3e
 
 
 
 
 
 
 
 
 
 
 
 
 
 
62a7741
1078d3e
 
 
5b0bc48
b6ab8aa
1078d3e
 
 
 
 
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
name: Tests
on:
  # check on push/merge to main, PRs, and manual triggers
  push:
    branches:
      - "main"
    paths:
      - '**.py'
      - 'requirements.txt'
  pull_request:
      paths:
       - '**.py'
       - 'requirements.txt'
  workflow_dispatch:

jobs:
  pre-commit:
    name: pre-commit
    runs-on: ubuntu-latest
    steps:
      - uses: actions/checkout@v3
      - uses: actions/setup-python@v4
        with:
          python-version: "3.9"
          cache: 'pip' # caching pip dependencies
      - uses: pre-commit/action@v3.0.0

  pytest:
    name: PyTest
    runs-on: ubuntu-latest
    strategy:
      fail-fast: false
      matrix:
        python_version: ["3.9", "3.10"]
    timeout-minutes: 10

    steps:
      - name: Check out repository code
        uses: actions/checkout@v3

      - name: Setup Python
        uses: actions/setup-python@v4
        with:
          python-version: ${{ matrix.python_version }}
          cache: 'pip' # caching pip dependencies

      - name: Install dependencies
        run: |
          pip3 install -U -e .
          pip3 install -r requirements-tests.txt

      - name: Run tests
        run: |
          pytest --ignore=tests/e2e/ tests/

  e2e-test:
    name: E2E Tests
    runs-on: [self-hosted, gpu]
    timeout-minutes: 20
    needs: [pre-commit, pytest]

    steps:
      - name: Check out repository code
        uses: actions/checkout@v3

      - name: Setup Python
        uses: actions/setup-python@v4
        with:
          python-version: "3.10"
#          cache: 'pip' # caching pip dependencies

      - name: Install dependencies
        run: |
          pip3 uninstall -y transformers accelerate
          pip3 install -U -e .[flash-attn]
          pip3 install -r requirements-tests.txt

      - name: Run e2e tests
        run: |
          pytest tests/e2e/