Skip to content

Commit a638d0d

Browse files
Create 'tests-openvino' extra dependency
1 parent 19e0a4f commit a638d0d

File tree

4 files changed

+4
-5
lines changed

4 files changed

+4
-5
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] transformers[testing]
46+
pip install .[openvino,openvino-tokenizers,diffusers,tests,tests-openvino] 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]
59+
pip install .[tests,tests-openvino]
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] transformers[testing]
45+
pip install .[openvino,tests,tests-openvino] transformers[testing]
4646
pip uninstall -y nncf
4747
4848
- if: ${{ matrix.transformers-version != 'latest' }}

setup.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -56,8 +56,6 @@
5656
"sentence-transformers",
5757
"open_clip_torch>=2.26.1",
5858
"peft",
59-
"librosa",
60-
"soundfile",
6159
]
6260

6361
QUALITY_REQUIRE = ["black~=23.1", "ruff==0.4.4"]
@@ -70,6 +68,7 @@
7068
"diffusers": ["diffusers"],
7169
"quality": QUALITY_REQUIRE,
7270
"tests": TESTS_REQUIRE,
71+
"tests-openvino": ["datasets[audio]>=1.4.0"]
7372
}
7473

7574
setup(

0 commit comments

Comments
 (0)