ENH: Support "extras" install, geomstats.__version__

geomstats/__about__.py

@@ -0,0 +1,15 @@
+__version__ = '1.11-dev'  # Remove -dev before releasing
+
+install_requires = [
+    'autograd',
+    'numpy>=1.14.1',
+    'scipy',
+    'matplotlib',
+    ]
+
+extras_require = {
+    'test': ['nose2', 'coverage', 'codecov'],
+    'tf': ['tensorflow>=1.8'],
+    'torch': ['torch==0.4.0'],
+    }
+extras_require['all'] = sum(extras_require.values(), [])

geomstats/__init__.py

@@ -1,3 +1,5 @@
+from .__about__ import __version__
+
 import geomstats.manifold
 import geomstats.euclidean_space
 import geomstats.hyperbolic_space

setup.py

@@ -1,11 +1,10 @@
 from setuptools import setup, find_packages
-
-with open('requirements.txt') as fp:
-        install_requires = fp.read()
+from geomstats.__about__ import __version__, install_requires, extras_require
 
 setup(name='geomstats',
-      version='1.11',
+      version=__version__,
       install_requires=install_requires,
+      extras_require=extras_require,
       description='Geometric statistics on manifolds',
       url='http://github.com/geomstats/geomstats',
       author='Nina Miolane',