setup.py

from setuptools import find_packages, setup

def _read_requirements_file(path: str):
    with open(path) as f:
        return list(
            map(
                lambda req: req.strip(),
                f.readlines(),
            )
        )

with open("README.md") as f:
    long_description = f.read()

setup(
    name="exohood-machine-learning",
    version="0.1.0",
    description="Machine Learning for Causal Inference",
    long_description=long_description,
    long_description_content_type="text/markdown",
    author="Exohood Protocol",
    url="https://github.com/exohood/exohood-machine-learning",
    classifiers=[
        "Programming Language :: Python :: 3 :: Only",
        "Programming Language :: Python :: 3.7",
        "Programming Language :: Python :: 3.8",
        "Programming Language :: Python :: 3.9",
    ],
    install_requires=_read_requirements_file("requirements.txt"),
    extras_require={
        "test": _read_requirements_file("requirements-dev.txt"),
        "ray": _read_requirements_file("requirements-ray.txt"),
    },
    packages=find_packages(
        include=["machine_learning", "machine_learning.*"],
        exclude=["tests*"],
    ),
    include_package_data=True,
    keywords="exohood-machine-learning",
)