from setuptools import setup, find_packages

extras = {
    'plots': ["seaborn", "jupyter"]
}

setup(
    name='selfrank',
    version='0.0.1',
    install_requires=[
        "numpy",
        "pandas",
        "requests",
        "matplotlib",
        "transformers[torch]",
        "rouge-score",
        "ibm-generative-ai",
        "langchain",
        'requests'
    ],
    extras_require=extras,
    packages=find_packages(include=['selfrank', 'selfrank.*']),
    include_package_data=False
)