Skip to content

Commit

Permalink
Remove tests with optimum installed from source (huggingface#819)
Browse files Browse the repository at this point in the history
* remove tests with optimum installed from source

* add test with lower version of supported transformers
  • Loading branch information
echarlaix committed Jul 12, 2024
1 parent db1c1ab commit 7238e74
Showing 1 changed file with 7 additions and 2 deletions.
9 changes: 7 additions & 2 deletions .github/workflows/test_openvino_basic.yml
Original file line number Diff line number Diff line change
Expand Up @@ -23,8 +23,12 @@ jobs:
# Testing lower and upper bound of supported Python versions
# This also ensures that the test fails if dependencies break for Python 3.7
python-version: ["3.8", "3.12"]
optimum: ['optimum', 'git+https://github.com/huggingface/optimum.git']
os: ["ubuntu-22.04", "windows-latest"]
transformers-version: ["4.42.*"]
include:
- transformers-version: "4.36.0"
python-version: "3.12"
os: "ubuntu-22.04"

runs-on: ${{ matrix.os }}

Expand All @@ -41,7 +45,8 @@ jobs:
# optimum or transformers to a specific version
# Install PyTorch CPU to prevent unnecessary downloading/installing of CUDA packages
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
pip install .[tests] openvino ${{ matrix.optimum}}
pip install transformers==${{ matrix.transformers-version }}
pip install .[tests] openvino
- name: Pip freeze
run: pip freeze
Expand Down

0 comments on commit 7238e74

Please sign in to comment.