Fix the build documentation job (#14788)

* Fix the build documentation job

* Fix install

* Address review comment
This commit is contained in:
Sylvain Gugger 2021-12-16 09:35:20 -05:00 committed by GitHub
parent 5061a9fd55
commit 12e1b4c6df
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -5,6 +5,7 @@ on:
branches:
- master
- doc-builder*
- v*-release
jobs:
build_and_package:
@ -46,7 +47,9 @@ jobs:
sudo apt-get -y update && sudo apt-get install -y libsndfile1-dev
pip install git+https://github.com/huggingface/doc-builder
pip install git+https://github.com/huggingface/transformers#egg=transformers[dev]
cd transformers
pip install .[dev]
cd ..
export TORCH_VERSION=$(python -c "from torch import version; print(version.__version__.split('+')[0])")
pip install torch-scatter -f https://data.pyg.org/whl/torch-${TORCH_VERSION}+cpu.html