# ---------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# ---------------------------------------------------------

import io
from setuptools import setup
from setuptools import find_packages

VERSION = '1.6'

NAME = 'inference-schema'

DESCRIPTION = 'This package is intended to provide a uniform schema for common machine learning applications, as ' \
              'well as a set of decorators that can be used to aid in web based ML prediction applications.'

DEPENDENCIES = [
    'python_dateutil>=2.5.3',
    'pytz>=2017.2',
    'wrapt>=1.11.1,<=1.12.1'
]

EXTRAS = {
    'numpy-support': ['numpy>=1.13.0'],
    'pandas-support': ['pandas>=0.20.2'],
    'pyspark-support': ['pyspark>=2.3.2']
}

CLASSIFIERS = [
    'Development Status :: 3 - Alpha',
    'Intended Audience :: Developers',
    'Intended Audience :: System Administrators',
    'Programming Language :: Python :: 3',
    'Programming Language :: Python :: 3.8',
    'Programming Language :: Python :: 3.9',
    'Programming Language :: Python :: 3.10',
]

with io.open('LICENSE.txt', 'r', encoding='utf-8') as f:
    LICENSE = f.read()

setup(
    name=NAME,
    version=VERSION,
    description=DESCRIPTION,
    classifiers=CLASSIFIERS,
    author='Microsoft Corp',
    license=LICENSE,
    install_requires=DEPENDENCIES,
    extras_require=EXTRAS,
    packages=find_packages(exclude=['tests', 'tests.*']),
    include_package_data=True,
    python_requires=">=3.7,<3.11"
)
