Enable CI for torch 1.10

This commit is contained in:
Fangjun Kuang 2021-11-05 22:27:56 +08:00
parent 84def93ec9
commit 3fbc4fd092

View File

@ -33,8 +33,8 @@ jobs:
# disable macOS test for now. # disable macOS test for now.
os: [ubuntu-18.04] os: [ubuntu-18.04]
python-version: [3.6, 3.7, 3.8, 3.9] python-version: [3.6, 3.7, 3.8, 3.9]
torch: ["1.8.1"] torch: ["1.8.1", "1.10"]
k2-version: ["1.9.dev20210919"] k2-version: ["1.9.dev20211101"]
fail-fast: false fail-fast: false
@ -83,7 +83,10 @@ jobs:
ls -lh ls -lh
export PYTHONPATH=$PWD:$PWD/lhotse:$PYTHONPATH export PYTHONPATH=$PWD:$PWD/lhotse:$PYTHONPATH
echo $PYTHONPATH echo $PYTHONPATH
pytest ./test pytest -v -s ./test
# runt tests for conformer ctc
cd egs/librispeech/ASR/conformer_ctc
pytest -v -s
- name: Run tests - name: Run tests
if: startsWith(matrix.os, 'macos') if: startsWith(matrix.os, 'macos')
@ -93,8 +96,8 @@ jobs:
lib_path=$(python -c "from distutils.sysconfig import get_python_lib; print(get_python_lib())") lib_path=$(python -c "from distutils.sysconfig import get_python_lib; print(get_python_lib())")
echo "lib_path: $lib_path" echo "lib_path: $lib_path"
export DYLD_LIBRARY_PATH=$lib_path:$DYLD_LIBRARY_PATH export DYLD_LIBRARY_PATH=$lib_path:$DYLD_LIBRARY_PATH
pytest ./test pytest -v -s ./test
# runt tests for conformer ctc # runt tests for conformer ctc
cd egs/librispeech/ASR/conformer_ctc cd egs/librispeech/ASR/conformer_ctc
pytest pytest -v -s