setup.py 1.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051
  1. from setuptools import setup
  2. import setuptools.command.build_py
  3. def readme():
  4. with open('README.rst') as f:
  5. return f.read()
  6. class BuildTestCommand(setuptools.command.build_py.build_py):
  7. """Custom build command."""
  8. def run(self):
  9. # run tests using doctest
  10. import doctest
  11. from ipfml import image_processing
  12. from ipfml import metrics
  13. from ipfml import tf_model_helper
  14. doctest.testmod(image_processing)
  15. doctest.testmod(metrics)
  16. doctest.testmod(tf_model_helper)
  17. setuptools.command.build_py.build_py.run(self)
  18. setup(name='IPFML',
  19. version='0.0.7',
  20. description='Image Processing For Machine Learning',
  21. long_description=readme(),
  22. classifiers=[
  23. 'Development Status :: 3 - Alpha',
  24. 'License :: OSI Approved :: MIT License',
  25. 'Programming Language :: Python :: 3.6',
  26. 'Topic :: Scientific/Engineering :: Artificial Intelligence'
  27. ],
  28. url='https://gogs.univ-littoral.fr/jerome.buisine/IPFML',
  29. author='Jérôme BUISINE',
  30. author_email='jerome.buisine@univ-littoral.fr',
  31. license='MIT',
  32. packages=['ipfml'],
  33. install_requires=[
  34. 'matplotlib',
  35. 'numpy',
  36. 'Pillow',
  37. 'sklearn',
  38. 'scikit-image',
  39. 'scipy',
  40. ],
  41. cmdclass={
  42. 'build_py': BuildTestCommand,
  43. },
  44. zip_safe=False)