Add datasets + transformers + scipy to test deps (#279)
Add datasets + transformers to test deps
This commit is contained in:
parent
170af08e7f
commit
7e1b202d5e
|
@ -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: |
|
||||
|
|
2
setup.py
2
setup.py
|
@ -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 = [
|
||||
|
|
Loading…
Reference in New Issue