from setuptools import find_packages, setup

requirements = [
    "numpy>=1.16",
    "torch",
    "matplotlib",
    "stheno",
    "varz",
    "wbml",
    "backends",
    "backends-matrix",
    "plum-dispatch",
]

setup(
    packages=find_packages(exclude=["docs"]),
    python_requires=">=3.6",
    install_requires=requirements,
    include_package_data=True,
)
