Skip to content

Commit f6b73d0

Browse files
Remove tests-openvino extra requirement (#1059)
* Remove openvino-tests extra requirement * export LD_LIBRARY_PATH * Add tbb to tests requirements
1 parent 02835ce commit f6b73d0

File tree

4 files changed

+5
-4
lines changed

4 files changed

+5
-4
lines changed

.github/workflows/test_openvino.yml

+1-1
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ jobs:
4343
run: |
4444
pip install --upgrade pip
4545
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
46-
pip install .[openvino,openvino-tokenizers,diffusers,tests,tests-openvino] transformers[testing]
46+
pip install .[openvino,openvino-tokenizers,diffusers,tests] transformers[testing]
4747
4848
- if: ${{ matrix.transformers-version != 'latest' }}
4949
name: Downgrade Transformers and Accelerate

.github/workflows/test_openvino_full.yml

+1-1
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@ jobs:
5656
python -m pip install --upgrade pip
5757
# Install PyTorch CPU to prevent unnecessary downloading/installing of CUDA packages
5858
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
59-
pip install .[tests,tests-openvino]
59+
pip install .[tests]
6060
6161
- name: Install openvino-nightly
6262
if: ${{ matrix.openvino == 'ov-nightly' }}

.github/workflows/test_openvino_slow.yml

+1-1
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ jobs:
4242
run: |
4343
pip install --upgrade pip
4444
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
45-
pip install .[openvino,tests,tests-openvino] transformers[testing]
45+
pip install .[openvino,tests] transformers[testing]
4646
pip uninstall -y nncf
4747
4848
- if: ${{ matrix.transformers-version != 'latest' }}

setup.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -56,6 +56,8 @@
5656
"sentence-transformers",
5757
"open_clip_torch>=2.26.1",
5858
"peft",
59+
"datasets[audio]>=1.4.0",
60+
"tbb",
5961
]
6062

6163
QUALITY_REQUIRE = ["black~=23.1", "ruff==0.4.4"]
@@ -68,7 +70,6 @@
6870
"diffusers": ["diffusers"],
6971
"quality": QUALITY_REQUIRE,
7072
"tests": TESTS_REQUIRE,
71-
"tests-openvino": ["datasets[audio]>=1.4.0"],
7273
}
7374

7475
setup(

0 commit comments

Comments
 (0)