setup.py

from setuptools import setup, find_packages
from pathlib import Path

project_root = Path(__file__).resolve().parent

about = {}
version_path = project_root / 'entmoot' / '__version__.py'
with version_path.open() as f:
    exec(f.read(), about)

setup(
    name='entmoot',
    author=about['__author__'],
    author_email=about['__author_email__'],
    license=about['__license__'],
    version=about['__version__'],
    url='https://github.com/cog-imperial/entmoot',
    packages=find_packages(exclude=['tests','docs']),
    install_requires=[
        'numpy>=1.18.4',
        'scikit-learn>=0.21.3',
        'pyYAML>=5.3.1',
        'lightgbm>=2.3.1'
    ],
    setup_requires=[
        'numpy>=1.18.4',
        'scikit-learn>=0.21.3',
        'pyYAML>=5.3.1',
        'lightgbm>=2.3.1'
    ]
)