Skip to content

Commit

Permalink
add test
Browse files Browse the repository at this point in the history
  • Loading branch information
echarlaix committed Oct 28, 2024
1 parent 6ebf667 commit f72f2bd
Showing 1 changed file with 12 additions and 3 deletions.
15 changes: 12 additions & 3 deletions .github/workflows/test_onnxruntime.yml
Original file line number Diff line number Diff line change
Expand Up @@ -17,8 +17,13 @@ jobs:
strategy:
fail-fast: false
matrix:
python-version: [3.9]
transformers-version: ["latest"]
os: [ubuntu-20.04, windows-2019, macos-13]
include:
os: ubuntu-20.04
transformers-version: "4.36.*"
os: ubuntu-20.04
transformers-version: "4.45.*"

runs-on: ${{ matrix.os }}
steps:
Expand All @@ -33,17 +38,21 @@ jobs:
- name: Checkout code
uses: actions/checkout@v4

- name: Setup Python ${{ matrix.python-version }}
- name: Setup Python
uses: actions/setup-python@v5
with:
python-version: ${{ matrix.python-version }}
python-version: "3.9"

- name: Install dependencies
run: |
pip install --upgrade pip
pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
pip install .[tests,onnxruntime]
- name: Install transformers ${{ matrix.transformers-version }}
if: ${{ matrix.transformers-version != 'latest' }} # already installed
run: pip install transformers==${{ matrix.transformers-version }}

- name: Test with pytest (in series)
working-directory: tests
run: |
Expand Down

0 comments on commit f72f2bd

Please sign in to comment.