setup.py

"""
TalkToEBM installation script.
"""

import setuptools

with open("README.md", "r", encoding="utf-8") as fh:
    long_description = fh.read()

# Read the version from the version file
version = {}
with open("t2ebm/version.py") as fp:
    exec(fp.read(), version)

setuptools.setup(
    name="t2ebm",
    version=version["__version__"],
    author="Sebastian Bordt, Ben Lengerich, Harsha Nori, Rich Caruana",
    author_email="sbordt@posteo.de",
    description="A Natural Language Interface to Explainable Boosting Machines",
    long_description=long_description,
    long_description_content_type="text/markdown",
    url="https://github.com/interpretml/TalkToEBM",
    classifiers=[
        "Programming Language :: Python :: 3",
        "License :: OSI Approved :: MIT License",
        "Operating System :: OS Independent",
    ],
    package_dir={"": "."},
    packages=["t2ebm"],
    python_requires=">=3.6",
    install_requires=[
        "numpy>=1.14.0,<2.0.0",
        "matplotlib",
        "tiktoken",
        "openai>=1.8.0",
        "tenacity",
        "scipy",
        "interpret",
    ],
)