import os

import setuptools
from pipelines import __version__ as version

with open("README.md", "r") as f:
    readme = f.read()


required_packages = [
    "pip==22.1.2",
    "boto3==1.24.22",
    "shap==0.41.0",
    "imbalanced-learn==0.9.1",
    "werkzeug==2.0.3",
    "pandas==1.2.4",
    "numpy==1.22.0",
    "xgboost==1.5.2",
    "scikit-learn==1.1.1",
    "sagemaker",
    "pre-commit",
    "pytest==6.2.5",
    "optuna==2.10.1",
    "scikeras==0.8.0",
    "tensorflow-cpu==2.9.3",
    "keras==2.9.0",
]
extras = {
    "test": [
        "black",
        "coverage",
        "flake8",
        "mock",
        "pydocstyle",
        "pytest",
        "pytest-cov",
        "sagemaker",
        "tox",
    ]
}
setuptools.setup(
    name=version.__title__,
    description=version.__description__,
    version=version.__version__,
    author=version.__author__,
    author_email=version.__author_email__,
    long_description=readme,
    long_description_content_type="text/markdown",
    url=version.__url__,
    license=version.__license__,
    packages=setuptools.find_packages(),
    include_package_data=True,
    python_requires=">=3.6",
    install_requires=required_packages,
    extras_require=extras,
    entry_points={
        "console_scripts": [
            "get-pipeline-definition=pipelines.get_pipeline_definition:main",
            "run-pipeline=pipelines.run_pipeline:main",
        ]
    },
    classifiers=[
        "Development Status :: 3 - Alpha",
        "Intended Audience :: Developers",
        "Natural Language :: English",
        "Programming Language :: Python",
        "Programming Language :: Python :: 3",
        "Programming Language :: Python :: 3.6",
        "Programming Language :: Python :: 3.7",
        "Programming Language :: Python :: 3.8",
    ],
)