from setuptools import setup, find_packages

setup(
    name='researcher',
    version='0.0.0',
    description='Open-source training recipe for reproducing Deepseek-R1.',
    author='',
    packages=find_packages(include=['researcher',]),
    install_requires=[
        'google-cloud-aiplatform',
        'latex2sympy2',
        'pylatexenc',
        'sentence_transformers',
        'tabulate',
        'flash_attn==2.7.3',
    ],
    classifiers=[
        'Programming Language :: Python :: 3',
        'License :: OSI Approved :: MIT License',
        'Operating System :: OS Independent',
    ],
    python_requires='>=3.9',
)