#!/usr/bin/env python

from setuptools import setup, find_packages

__version__ = None
exec(open('dltk/version.py').read())

test_require = ['pytest', 'pytest-flake8', 'pytest-cov', 'python-coveralls']

setup(name='dltk',
      version=__version__,
      description='Deep Learning Toolkit for Medical Image Analysis',
      author='DLTK contributors',
      url='https://dltk.github.io',
      packages=find_packages(exclude=['docs', 'contrib', 'data', 'examples']),
      keywords=['machine learning', 'tensorflow', 'deep learning',
                'biomedical imaging'],
      license='Apache License 2.0',
      classifiers=['Intended Audience :: Developers',
                   'Intended Audience :: Education',
                   'Intended Audience :: Science/Research',
                   'License :: OSI Approved :: Apache Software License',
                   'Operating System :: POSIX :: Linux',
                   'Operating System :: MacOS :: MacOS X',
                   'Operating System :: Microsoft :: Windows',
                   'Programming Language :: Python :: 2.7',
                   'Programming Language :: Python :: 3.4'],
      install_requires=['numpy>=1.14.0', 'scipy>=0.19.0', 'pandas>=0.19.0',
                        'matplotlib>=1.5.3', 'future>=0.16.0', 'xlrd>=1.1.0',
                        'scikit-image>=0.13.0', 'SimpleITK>=1.0.0',
                        'jupyter>=1.0.0', 'argparse'],
      tests_require=test_require,
      extras_require={'docs': ['sphinx==1.5.6', 'sphinx-rtd-theme',
                               'recommonmark', 'sphinx-autobuild',
                               'sphinxcontrib-versioning'],
                      'tests': test_require}
      )

print("\nWelcome to DLTK!")
print("If any questions please visit documentation page "
      "https://dltk.github.io/dltk")
print("or join community chat on https://gitter.im/DLTK/DLTK")


try:
    import tensorflow
except ImportError:
    print('We did not find TensorFlow on your system. Please install it via '
          '`pip install tensorflow-gpu` if you have a CUDA-enabled GPU or with '
          '`pip install tensorflow` without GPU support.')