import codecs
import os

from setuptools import find_packages, setup

install_requires = [
    "maite==0.3.4",
    "adversarial-robustness-toolbox==1.16.0",
    "scikit-learn>=0.22.2,<1.2.0",
    "six",
    "setuptools",
    "tqdm",
]


def read(rel_path):
    here = os.path.abspath(os.path.dirname(__file__))
    with codecs.open(os.path.join(here, rel_path), "r", encoding="utf-8") as fp:
        return fp.read()


def get_version(rel_path):
    for line in read(rel_path).splitlines():
        if line.startswith("__version__"):
            delim = '"' if '"' in line else "'"
            return line.split(delim)[1]
    raise RuntimeError("Unable to find version string.")


setup(
    name="hardened-extension-adversarial-robustness-toolbox",
    version=get_version("src/heart/__init__.py"),
    description="Extension for ART compatible with MAITE.",
    author="IBM",
    author_email="<email>",
    maintainer="IBM",
    maintainer_email="<email>",
    license="MIT",
    install_requires=install_requires,
    include_package_data=True,
    python_requires=">=3.9,<3.11",
)