setup.py

from setuptools import setup, find_packages

setup(name='limbus',
      version='0.1.6',
      description='High level interface to create Pytorch Graphs.',
      long_description=open('README.md').read(),
      long_description_content_type='text/markdown',
      author='Luis Ferraz',
      url='https://github.com/kornia/limbus',
      install_requires=[
          'typeguard<3.0.0',
      ],
      extras_require={
          'dev': [
              'pytest',
              'pytest-flake8',
              'pytest-cov',
              'pytest-mypy',
              'pytest-pydocstyle',
              'pytest-asyncio',
              'mypy',  # TODO: check if we can remove the deps without pytest-*
              'pydocstyle',
              'flake8<5.0.0',  # last versions of flake8 are not compatible with pytest-flake8==1.1.1 (lastest version)
              'pep8-naming',
          ],
          'components': [
              'limbus-components'
          ],
          'widgets': [
              'kornia',
              'torch',
              'numpy',
              'visdom',
              'opencv-python',
          ]
      },
      packages=find_packages(where='.'),
      package_dir={'': '.'},
      package_data={'': ['*.yml']},
      include_package_data=True
      )