from setuptools import setup, find_packages

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

setup(
    name='bilbystats',
    version='0.2.0',
    license='MIT',
    author='Samuel DAVENPORT',
    author_email='12sdavenport@gmail.com',
    url='https://github.com/sjdavenport/bilbystats/',
    download_url='https://github.com/bilbyai/bilbystats/',
    description='Python Packages of functions for performing stats for bilby',
    long_description=long_description,
    long_description_content_type='text/markdown',
    keywords='LLMs, Transformers',
    packages=find_packages(),
    include_package_data=True,
    package_data={
        "bilbystats": ["data/*.csv", "data/*.json", "data/*.parquet", "data/prompts/*.txt", "data/entities/conll2003/*.txt", "data/*.html", "data/entities/conll2003/*.parquet"],
        "bilbystats.defaults": ["config.yaml", "*.env"],
        "bilbystats.defaults.training_defaults": ["*.env"],
    },
    install_requires=[
        'numpy',
        'matplotlib',
        'scipy',
        'transformers>=4.48.0',
        'datasets',
        'evaluate',
        'torch',
        'pandas',
        'openai',
        'dotenv',
        'anthropic',
        'ollama',
        'google-genai',
        'tiktoken',
        'transformers[torch]',
        'seaborn',
        'google-cloud-bigquery',
        'db-dtypes',
        'ipywidgets',
        'ipykernel',
        'google-cloud-bigquery-storage',
        'nbformat',
        'nbclient',
        'nasdaq-data-link',
        'wandb',
        'sentence-transformers',
        'gliner',
        'sacrebleu',
        'bert-score',
        'pytorch-lightning>=2.0.0',
        'groq',
        'seqeval',
        'nltk',
    ],
    extras_require={
        "dev": [
            'yfinance',
            'jupyter',
            'snownlp',
            'spacy',
            'geopandas',
            'dspy',
            'litellm',
            'unbabel-comet>=2.2.0',
            'spyder-kernels==3.0.*',
            'beautifulsoup4',
            'scikit-learn',
            'google-api-python-client',
            'trafilatura',
            'selenium',
            'webdriver_manager',
            'openpyxl'
        ],
    },
    python_requires='>=3.9',
)
