setup.py

import setuptools

from setuptools import find_packages

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

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


setuptools.setup(
    name="tabmemcheck",
    version=version["__version__"],
    author="Sebastian Bordt, Harsha Nori, Rich Caruana",
    author_email="sbordt@posteo.de",
    description="Testing Language Models for Memorization of Tabular Data",
    long_description=long_description,
    long_description_content_type="text/markdown",
    url="https://github.com/interpretml/LLM-Tabular-Memorization-Checker",
    classifiers=[
        "Programming Language :: Python :: 3",
        "License :: OSI Approved :: MIT License",
        "Operating System :: OS Independent",
    ],
    package_dir={"": "."},
    packages=find_packages(),
    package_data={
        "": ["*.yaml", "*.csv"],
    },
    include_package_data=True,
    python_requires=">=3.9",
    install_requires=[
        "numpy",
        "matplotlib",
        "scikit-learn",
        "xgboost",
        "scipy",
        "pandas>=2.1.0",
        "tiktoken",
        "openai>=1.3.3",
        "tenacity",
        "jellyfish",
        "pyyaml",
    ],
)