# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
# SPDX-License-Identifier: Apache-2.0
import os
from setuptools import find_packages, setup

INSTALL_REQUIRES = ["boto3", "pyarrow", "pandas", "s3fs", "numpy", "scikit-learn", "pyfunctional"]

EXTRAS_REQUIRE = {
    "test": [
        "tox",
        "flake8",
        "black == 22.3.0",
        "mock",
        "pre-commit",
        "pytest",
        "pytest-pspec",
        "sphinx",
        "coverage",
        "nbconvert",
        "jupyter",
        "jupyter_contrib_nbextensions",
        "seaborn",
        "mypy",
        "lxml",
    ]
}

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

data_files = ["COMMIT_HASH"] if os.path.exists("./src/smclarify/COMMIT_HASH") else []

setup(
    name="smclarify",
    version="0.5",
    packages=find_packages("src"),
    package_dir={"": "src"},
    package_data={"smclarify": data_files},
    long_description=LONG_DESCRIPTION,
    long_description_content_type="text/markdown",
    python_requires=">=3.7",
    install_requires=INSTALL_REQUIRES,
    extras_require=EXTRAS_REQUIRE,
    classifiers=[
        "Development Status :: 1 - Planning",
        "Intended Audience :: Developers",
        "Natural Language :: English",
        "Programming Language :: Python",
        "Programming Language :: Python :: 3.7",
        "Programming Language :: Python :: 3.8",
        "Programming Language :: Python :: 3.9",
        "Programming Language :: Python :: 3.10",
    ],
)
