Commit 5123ded4 authored by Will Frey's avatar Will Frey Committed by Francisco Massa
Browse files

Add scipy as a dependency to setup.py (#675)

* Add scipy as a dependency to setup.py

`scipy` is imported in both the `torchvision.datasets.svhn` module and the `torchvision.models.inception` module.

* Update setup.py

Make scipy optional. It can be installed with `pip install torchvision[scipy]`.
parent 2d493d6a
......@@ -64,4 +64,7 @@ setup(
zip_safe=True,
install_requires=requirements,
extras_require={
"scipy": ["scipy"],
},
)
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