from setuptools import setup

setup(
    name='jointist',
    version='0.0.1',
    description='pop music transcription',
    author='SAMI',
    url="",
    author_email='',
    license='',
    packages=['jointist'],
    include_package_data=True,
    #TODO
    install_requires=[
        'torch==1.7.1',
        'librosa==0.8.0',  # specify the version!
        'mir_eval==0.6',
        'h5py==2.10.0',
        'tqdm==4.56.0',
        'mido==1.2.9',
        'pytorch_lightning==1.2.1',
        'numpy==1.18.5',
        'sox==1.4.1',
        'pretty_midi==0.2.9',
        'pyfluidsynth==1.3.0',
        'torchlibrosa==0.0.9',
        'matplotlib==3.3.4',
        'pandas==1.2.1',
        'auditok',
        'einops==0.3.0',
        'black==20.8b1'
    ],
    keywords='',
    zip_safe=False
)
