Unverified Commit 7e1b202d authored by Anton Lozhkov's avatar Anton Lozhkov Committed by GitHub
Browse files

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

Add datasets + transformers to test deps
parent 170af08e
......@@ -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: |
......
......@@ -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 = [
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment