from pathlib import Path

from setuptools import find_packages, setup

if __name__ == "__main__":
    with Path(Path(__file__).parent, "README.md").open(encoding="utf-8") as file:
        long_description = file.read()

    with open("requirements.txt") as f:
        REQUIREMENTS = f.read().splitlines()

    # with open("requirements-eval.txt") as f:    
    #     EVAL = f.read().splitlines()

    with open("requirements-training.txt") as f:
        TRAINING = f.read().splitlines()

    setup(
        name="open_flamingo",
        packages=find_packages(),
        include_package_data=True,
        version="2.0.1",
        license="MIT",
        description="An open-source framework for training large multimodal models",
        long_description=long_description,
        long_description_content_type="text/markdown",
        data_files=[(".", ["README.md"])],
        keywords=["machine learning"],
        install_requires=REQUIREMENTS,
        extras_require={
            # "eval": EVAL,
            "training": TRAINING,
            # "all": list(set(REQUIREMENTS + EVAL + TRAINING)),
            "all": list(set(REQUIREMENTS + TRAINING)),
        },
        classifiers=[
            "Development Status :: 4 - Beta",
            "Intended Audience :: Developers",
            "Topic :: Scientific/Engineering :: Artificial Intelligence",
            "License :: OSI Approved :: MIT License",
            "Programming Language :: Python :: 3.9",
        ],
    )