from setuptools import setup, find_packages

setup(
    name='rllm',
    version='0.0.0',
    description='Distributed Post-Training RL Library for LLMs',
    author='Agentica Team',
    packages=find_packages(include=['rllm',]),
    install_requires=[
        'torch==2.4.0',
        'deepspeed',
        'dm-tree',
        'gdown',
        'google-cloud-aiplatform',
        'latex2sympy2',
        'pybind11',
        'pylatexenc',
        'pytest',
        'sentence_transformers',
        'sortedcontainers',
        'tabulate',
        'torchmetrics',
        'flash-attn',
        'gym',
        'selenium',
        'PyMuPDF',
        'nltk',
        'browsergym',
        'gradio',
        'fire',
        'vertexai',
        'e2b_code_interpreter',
        'firecrawl',
        'wandb',
    ],
    classifiers=[
        'Programming Language :: Python :: 3',
        'License :: OSI Approved :: MIT License',
        'Operating System :: OS Independent',
    ],
    python_requires='>=3.9',
)