from distutils.core import setup

setup(
    name='attention',
    version='0.1.0',
    author='tllake',
    author_email='thom.l.lake@gmail.com',
    packages=['attention'],
    description='An attention function for PyTorch.',
    long_description=open('README.md').read())
add setup.py script