# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License.

from setuptools import setup
from os import path


UTILS_CV = "utils_cv"  # Utility folder name
README = path.join(UTILS_CV, "README.md")
exec(open(path.join(UTILS_CV, "__init__.py")).read())

with open(README, encoding="utf-8") as f:
    long_description = f.read()

setup(
    name="utils_cv",
    version=__version__,
    description="Computer Vision Utilities",
    long_description=long_description,
    long_description_content_type="text/markdown",
    url="https://github.com/microsoft/ComputerVision",
    project_urls={
        "Bug Tracker": "https://github.com/microsoft/ComputerVision/issues",
        "Source Code": "https://github.com/microsoft/ComputerVision",
        "Documentation": "https://github.com/microsoft/ComputerVision",
    },
    author="CVDev Team at Microsoft",
    classifiers=[
        "Development Status :: 2-Pre-Alpha",
        "Intended Audience :: Developers",
        "Intended Audience :: Science/Research",
        "License :: OSI Approved :: MIT License",
        "Programming Language :: Python :: 3",
        "Topic :: Scientific/Engineering",
        "Topic :: Software Development :: Libraries :: Python Modules",
    ],
    include_package_data=True,
    # Make utils_cv pip-installable without environment.yml.
    # * But keep environment.yml unchanged util final decision to
    #   incorporate utils_cv into environment.yml.
    # * Make sure to sync with environment.yml if any dependencies or
    #   versions changed.
    install_requires=[
        "azureml-sdk[notebooks,contrib]>=1.0.30",  # requires ipykernel, papermill, jupyter-core, jupyter-client
        "bqplot",
        "fastai==1.0.48",  # requires pytorch, torchvision, nvidia-ml-py3
        "scikit-learn>=0.19.1",
    ],
    keywords=", ".join(
        [
            "computer vision",
            "deep learning",
            "convolutional neural network",
            "image classification",
            "image similarity",
            "data science",
            "artificial intelligence",
            "machine learning",
            "gpu",
        ]
    ),
    packages=[UTILS_CV],
    python_requires=">=3.6",
)