setup.py 1.6 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344
  1. from pkg_resources import parse_requirements
  2. from setuptools import setup
  3. import codecs
  4. import re
  5. import os
  6. here = os.path.abspath(os.path.dirname(__file__))
  7. with open('requirements.txt') as requirements_file:
  8. install_requires = [str(requirement) for requirement in parse_requirements(requirements_file)]
  9. # loading version from setup.py
  10. with codecs.open(os.path.join(here, 'hivemind/__init__.py'), encoding='utf-8') as init_file:
  11. version_match = re.search(r"^__version__ = ['\"]([^'\"]*)['\"]", init_file.read(), re.M)
  12. version_string = version_match.group(1)
  13. setup(
  14. name='hivemind',
  15. version=version_string,
  16. description='',
  17. long_description='',
  18. author='Learning@home authors',
  19. author_email='mryabinin@hse.ru',
  20. url="https://github.com/learning-at-home/hivemind",
  21. packages=['hivemind'],
  22. license='MIT',
  23. install_requires=install_requires,
  24. classifiers=[
  25. 'Development Status :: 4 - Beta',
  26. 'Intended Audience :: Developers',
  27. 'Intended Audience :: Science/Research',
  28. 'License :: OSI Approved :: MIT License',
  29. 'Programming Language :: Python :: 3',
  30. 'Programming Language :: Python :: 3.8',
  31. 'Topic :: Scientific/Engineering',
  32. 'Topic :: Scientific/Engineering :: Mathematics',
  33. 'Topic :: Scientific/Engineering :: Artificial Intelligence',
  34. 'Topic :: Software Development',
  35. 'Topic :: Software Development :: Libraries',
  36. 'Topic :: Software Development :: Libraries :: Python Modules',
  37. ],
  38. # What does your project relate to?
  39. keywords='pytorch, deep learning, machine learning, gpu, distributed computing',
  40. )