from setuptools import setup, find_packages

VERSION = "0.1.2"
DESCRIPTION = "A standalone package to do the vitis quantization step."

REQUIRED_PACKAGES = [
    "tensorflow>=2.7.0",
    "numpy>=1.22.1",
    "tensorflow-model-optimization>=0.7.0",
]

setup(
    name="vitis-quantizer",
    version=VERSION,
    description=DESCRIPTION,
    url="https://github.com/CruxML/vitis-quantizer",
    author="CruxML",
    author_email="stephen.tridgell@cruxml.com",
    install_requires=REQUIRED_PACKAGES,
    long_description=open("README.md").read(),
    download_url=f"https://github.com/CruxML/vitis-quantizer/archive/refs/tags/{VERSION}.tar.gz",
    packages=find_packages(),
    license="Apache 2.0",
    include_package_data=True,
    keywords="vitis quantize machine learning xilinx",
    long_description_content_type='text/markdown',
)
