From 7238e7437a51300df78df61bc201a90115110fda Mon Sep 17 00:00:00 2001 From: Ella Charlaix <80481427+echarlaix@users.noreply.github.com> Date: Fri, 12 Jul 2024 11:50:27 +0200 Subject: [PATCH] Remove tests with optimum installed from source (#819) * remove tests with optimum installed from source * add test with lower version of supported transformers --- .github/workflows/test_openvino_basic.yml | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/.github/workflows/test_openvino_basic.yml b/.github/workflows/test_openvino_basic.yml index 141b94425e..dc192852ec 100644 --- a/.github/workflows/test_openvino_basic.yml +++ b/.github/workflows/test_openvino_basic.yml @@ -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 }} @@ -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