setup.py 1.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455
  1. import setuptools
  2. import codecs
  3. import os.path
  4. with open("README.md", "r") as fh:
  5. long_description = fh.read()
  6. def read(rel_path):
  7. here = os.path.abspath(os.path.dirname(__file__))
  8. with codecs.open(os.path.join(here, rel_path), 'r') as fp:
  9. return fp.read()
  10. def get_version(rel_path):
  11. for line in read(rel_path).splitlines():
  12. if line.startswith('__version__'):
  13. delim = '"' if '"' in line else "'"
  14. return line.split(delim)[1]
  15. else:
  16. raise RuntimeError("Unable to find version string.")
  17. setuptools.setup(
  18. name="avalanche-lib", # Replace with your own username
  19. version=get_version("avalanche/__init__.py"),
  20. author="ContinualAI",
  21. author_email="contact@continualai.org",
  22. description="Avalanche: a Comprehensive Framework for Continual Learning "
  23. "Research",
  24. long_description=long_description,
  25. long_description_content_type="text/markdown",
  26. url="https://github.com/vlomonaco/avalanche",
  27. packages=setuptools.find_packages(),
  28. classifiers=[
  29. "Programming Language :: Python :: 3",
  30. "License :: OSI Approved :: MIT License",
  31. "Operating System :: OS Independent",
  32. ],
  33. python_requires='>=3.6,<3.10',
  34. install_requires=[
  35. 'typing-extensions',
  36. 'psutil',
  37. 'gputil',
  38. 'scikit-learn',
  39. 'matplotlib',
  40. 'numpy',
  41. 'pytorchcv',
  42. 'quadprog',
  43. 'wandb',
  44. 'tensorboard',
  45. 'pycocotools',
  46. 'gdown',
  47. 'pandas',
  48. 'scikit-image',
  49. 'ipykernel'
  50. ]
  51. )