Skip to content

Commit 1ccac0e

Browse files
Remove unnecessary installations for linux CPU test (#4037)
* Remove unnecessary installations * Set CTC Decoder skipping * Remove pillow install * Make the test env variable lower case * Add extra module skips
1 parent 6a044df commit 1ccac0e

File tree

2 files changed

+6
-10
lines changed

2 files changed

+6
-10
lines changed

.github/scripts/unittest-linux/install.sh

Lines changed: 2 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -98,14 +98,6 @@ if [[ "$(python --version)" = *3.9* || "$(python --version)" = *3.10* ]]; then
9898
fi
9999
(
100100
set -x
101-
conda install -y -c conda-forge ${NUMBA_DEV_CHANNEL} libvorbis parameterized 'requests>=2.20'
102-
pip install SoundFile coverage pytest pytest-cov scipy expecttest unidecode inflect Pillow sentencepiece pytorch-lightning 'protobuf<4.21.0' demucs tinytag pyroomacoustics flashlight-text git+https://github.com/kpu/kenlm
103-
104-
# TODO: might be better to fix the single call to `pip install` above
105-
pip install pillow scipy "numpy>=1.26"
101+
conda install -y -c conda-forge ${NUMBA_DEV_CHANNEL} parameterized 'requests>=2.20'
102+
pip install coverage pytest pytest-cov scipy expecttest scipy "numpy>=1.26"
106103
)
107-
# Install fairseq
108-
git clone https://github.com/pytorch/fairseq
109-
cd fairseq
110-
git checkout e47a4c8
111-
pip install .

.github/scripts/unittest-linux/run_test.sh

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,10 @@ fi
2929
)
3030

3131
(
32+
export TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_CTC_DECODER=true
33+
export TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_MOD_unidecode=true
34+
export TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_MOD_inflect=true
35+
export TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_MOD_pytorch_lightning=true
3236
cd test
3337
pytest torchaudio_unittest -k "not backend and not (sox and not compatibility) and not /io/ and not prototype and not ffmpeg and not fairseq and not hdemucs and not (torchscript and rnnt) and not torchscript_consistency"
3438
)

0 commit comments

Comments
 (0)