"""Install ProtoTorch.""" from setuptools import setup from setuptools import find_packages PROJECT_URL = 'https://github.com/si-cim/prototorch' DOWNLOAD_URL = 'https://github.com/si-cim/prototorch.git' with open('README.md', 'r') as fh: long_description = fh.read() setup(name='prototorch', version='0.1.1-dev0', description='Highly extensible, GPU-supported ' 'Learning Vector Quantization (LVQ) toolbox ' 'built using PyTorch and its nn API.', long_description=long_description, long_description_content_type='text/markdown', author='Jensun Ravichandran', author_email='jjensun@gmail.com', url=PROJECT_URL, download_url=DOWNLOAD_URL, license='MIT', install_requires=[ 'torch>=1.3.1', 'torchvision>=0.5.0', 'numpy>=1.9.1', ], extras_require={ 'datasets': ['requests'], 'examples': [ 'sklearn', 'matplotlib', ], 'tests': ['pytest'], }, classifiers=[ 'Development Status :: 2 - Pre-Alpha', 'Environment :: Console', 'Intended Audience :: Developers', 'Intended Audience :: Education', 'Intended Audience :: Science/Research', 'License :: OSI Approved :: MIT License', 'Programming Language :: Python :: 3.6', 'Programming Language :: Python :: 3.7', 'Programming Language :: Python :: 3.8', 'Operating System :: OS Independent', 'Topic :: Scientific/Engineering :: Artificial Intelligence', 'Topic :: Software Development :: Libraries', 'Topic :: Software Development :: Libraries :: Python Modules' ], packages=find_packages())