import imp  # pylint: disable=deprecated-module

import setuptools

# Additional requirements for testing.
testing_require = [
    'mock',
    'pytest-xdist',
    'pytype',
]

setuptools.setup(
    name='neural_testbed',
    description=(
        'Neural testbed. '
        'A library for evaluating probabilistic inference in neural networks.'),
    long_description=open('README.md').read(),
    long_description_content_type='text/markdown',
    version=imp.load_source('_metadata',
                            'neural_testbed/_metadata.py').__version__,
    keywords='probabilistic-inference python machine-learning',
    packages=setuptools.find_packages(),
    classifiers=[
        'Development Status :: 3 - Alpha',
        'Environment :: Console',
        'Intended Audience :: Science/Research',
        'License :: OSI Approved :: Apache Software License',
        'Operating System :: POSIX :: Linux',
        'Operating System :: Microsoft :: Windows',
        'Operating System :: MacOS :: MacOS X',
        'Programming Language :: Python :: 3.7',
        'Programming Language :: Python :: 3.8',
        'Programming Language :: Python :: 3.9',
    ],
)
