setup.py

import sys

import setuptools

# READ README.md for long description on PyPi.
try:
    long_description = open("README.md", encoding="utf-8").read()
except Exception as e:
    sys.stderr.write(f"Failed to read README.md:\n  {e}\n")
    sys.stderr.flush()
    long_description = ""

setuptools.setup(
    name="bayesian_sde_solver",
    author="Yvann Le Fay, Adrien Corenflos",
    description="Bayesian SDE solver",
    long_description=long_description,
    version="0.1",
    packages=setuptools.find_packages(),
    install_requires=[
        "jax>=0.3.25",
        "jaxlib>=0.3.25",
        "pytest",
        "numpy>=1.24.3",
        "tqdm"
    ],
    long_description_content_type="text/markdown",
    keywords="probabilistic state space bayesian statistics sampling algorithms sde",
    license="MIT",
    license_files=("LICENSE",),
)