Add datasets + transformers + scipy to test deps (#279)

Add datasets + transformers to test deps
This commit is contained in:
Anton Lozhkov 2022-08-30 20:19:21 +02:00 committed by GitHub
parent 170af08e7f
commit 7e1b202d5e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 1 additions and 2 deletions

View File

@ -40,7 +40,6 @@ jobs:
python -m pip uninstall -y torch torchvision torchtext
python -m pip install torch --extra-index-url https://download.pytorch.org/whl/cu116
python -m pip install -e .[quality,test]
python -m pip install scipy transformers
- name: Environment
run: |

View File

@ -168,7 +168,7 @@ extras = {}
extras["quality"] = ["black==22.3", "isort>=5.5.4", "flake8>=3.8.3", "hf-doc-builder"]
extras["docs"] = ["hf-doc-builder"]
extras["training"] = ["accelerate", "datasets", "tensorboard", "modelcards"]
extras["test"] = ["pytest", "pytest-timeout", "pytest-xdist"]
extras["test"] = ["datasets", "pytest", "pytest-timeout", "pytest-xdist", "scipy", "transformers"]
extras["dev"] = extras["quality"] + extras["test"] + extras["training"] + extras["docs"]
install_requires = [