From 040863f6e6986a8d4d3ebaff7fb0671db7f7aadf Mon Sep 17 00:00:00 2001 From: Miroslav Batchkarov Date: Mon, 30 Mar 2015 09:34:31 +0100 Subject: [PATCH] added optional requirements --- setup.py | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 deletions(-) diff --git a/setup.py b/setup.py index ce86b389d6f..001dd563bc1 100644 --- a/setup.py +++ b/setup.py @@ -13,11 +13,15 @@ url='https://github.com/fchollet/keras', license='MIT', packages=[ - 'keras', - 'keras.layers', - 'keras.preprocessing', - 'keras.datasets', - 'keras.utils', + 'keras', + 'keras.layers', + 'keras.preprocessing', + 'keras.datasets', + 'keras.utils', ], - install_requires=['numpy', 'scipy', 'theano'] + install_requires=['numpy', 'scipy', 'theano'], + extras_require={ + 'images': ['pil'], # working with images + 'CNNs': ['cudnn-python-wrappers'], # working with CNNs + } ) \ No newline at end of file