Merge pull request #29 from smartkiwi/master

Install package dependencies automatically

Merge pull request #29 from smartkiwi/master
Install package dependencies automatically
e869fcbe · Soumith Chintala · GitHub · 048f6f50 · 2fb84474 · e869fcbe
Commit e869fcbe authored Jan 18, 2017 by Soumith Chintala Committed by GitHub Jan 18, 2017
Show whitespace changes
Inline Side-by-side

Showing with 9 additions and 0 deletions

setup.py setup.py +9 -0

No files found.
--- a/setup.py
+++ b/setup.py
@@ -9,6 +9,14 @@ readme = open('README.rst').read()

 VERSION = '0.1.6'

+# same as ./requirements.txt
+requirements = [
+    'numpy',
+    'pillow',
+    'six',
+    'torch',
+]
+
 setup(
    # Metadata
    name='torchvision',
@@ -24,4 +32,5 @@ setup(
    packages=find_packages(exclude=('test',)),

    zip_safe=True,
+    install_requires=requirements,
 )