Skip to content

Commit

Permalink
ci: optimize installation by using CPU only torch
Browse files Browse the repository at this point in the history
  • Loading branch information
joanise committed Jan 25, 2024
1 parent 5b72112 commit 328d0ab
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion .github/workflows/test.yml
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,11 @@ jobs:
python-version: "3.8"
cache: "pip"
- name: Install dependencies and package
run: pip install -e . mypy
run: |
# Get the CPU only torch and torchaudio packages first
pip install $(grep torch requirements.txt | sed 's/$/+cpu/') --find-links https://download.pytorch.org/whl/torch_stable.html
# Install everything else
pip install -e . mypy
- name: Minimal test, --help should work
run: ctc-segmenter --help
- name: Code quality test, mypy should pass
Expand Down

0 comments on commit 328d0ab

Please sign in to comment.