from setuptools import setup, find_packages
from pathlib import Path

long_description = (Path(__file__).parent / "README.md").read_text()

setup(
    name="scikit-mdn",
    description="Mixture Density Networks in scikit-learn",
    long_description=long_description,
    long_description_content_type="text/markdown",
    author="Vincent D. Warmerdam",
    version="0.0.1",
    packages=find_packages(),
    install_requires=["scikit-learn", "torch"],
    extras_require={
        "dev": ["pytest"],
    },
)
