import pathlib
from setuptools import setup, find_packages


HERE = pathlib.Path(__file__).parent

README = (HERE / "README.md").read_text()

setup(
    name='tacklebox',
    version='1.0.2',
    description='Improved handling of PyTorch module hooks',
    long_description=README,
    long_description_content_type='text/markdown',
    url='https://github.com/IsaacRe/tacklebox',
    author='Isaac Rehg',
    author_email='isaacrehg@gmail.com',
    classifiers=[
        "Programming Language :: Python :: 3",
        "Programming Language :: Python :: 3.7"
    ],
    packages=find_packages(exclude=("tests",)),
    include_package_data=False,
    install_requires=[
        'torch',
        'tqdm',
    ],
)