Skip to content

Commit c2c68d6

Browse files
Slow test: run on pre-release; run full
1 parent 7d6a949 commit c2c68d6

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

.github/workflows/test_openvino_slow.yml

+2-1
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,7 @@ jobs:
4343
pip install --upgrade pip
4444
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
4545
pip install .[openvino,tests] transformers[testing]
46+
pip install --pre -U openvino openvino-tokenizers --extra-index-url https://storage.openvinotoolkit.org/simple/wheels/pre-release
4647
pip uninstall -y nncf
4748
4849
- if: ${{ matrix.transformers-version != 'latest' }}
@@ -71,7 +72,7 @@ jobs:
7172

7273
- name: Test with Pytest (slow)
7374
run: |
74-
pytest tests/openvino -m "run_slow" --durations=0
75+
pytest tests/openvino --durations=0
7576
env:
7677
RUN_SLOW: 1
7778
HF_HUB_READ_TOKEN: ${{ secrets.HF_HUB_READ_TOKEN }}

setup.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -63,8 +63,8 @@
6363
QUALITY_REQUIRE = ["black~=23.1", "ruff==0.4.4"]
6464

6565
EXTRAS_REQUIRE = {
66-
"nncf": ["nncf>=2.14.0"],
67-
"openvino": ["nncf>=2.14.0", "openvino>=2024.5.0", "openvino-tokenizers>=2024.5.0"],
66+
"nncf": ["nncf@git+https://github.com/openvinotoolkit/nncf.git@release_v2150"],
67+
"openvino": ["nncf@git+https://github.com/openvinotoolkit/nncf.git@release_v2150", "openvino>=2024.5.0", "openvino-tokenizers>=2024.5.0"],
6868
"neural-compressor": ["neural-compressor[pt]>3.0", "accelerate", "transformers<4.46"],
6969
"ipex": ["intel-extension-for-pytorch>=2.4", "transformers>4.46,<4.48", "accelerate"],
7070
"diffusers": ["diffusers"],

0 commit comments

Comments
 (0)