Mirrors
/
diffusers
mirror of https://github.com/ShivamShrirao/diffusers.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
.github/workflows/push_tests.yml vendored
View File

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

2
setup.py
View File

@ -168,7 +168,7 @@ extras = {}
extras["quality"] = ["black==22.3", "isort>=5.5.4", "flake8>=3.8.3", "hf-doc-builder"] extras["quality"] = ["black==22.3", "isort>=5.5.4", "flake8>=3.8.3", "hf-doc-builder"]
extras["docs"] = ["hf-doc-builder"] extras["docs"] = ["hf-doc-builder"]
extras["training"] = ["accelerate", "datasets", "tensorboard", "modelcards"] 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"] extras["dev"] = extras["quality"] + extras["test"] + extras["training"] + extras["docs"]
install_requires = [ install_requires = [