forked from asteroid-team/asteroid
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
48 lines (45 loc) · 1.51 KB
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
from setuptools import setup, find_packages
asteroid_version = "0.3.0"
with open("README.md", encoding='utf-8') as fh:
long_description = fh.read()
setup(
name='asteroid',
version=asteroid_version,
author='Manuel Pariente',
author_email='[email protected]',
url="https://github.com/mpariente/asteroid",
description='PyTorch-based audio source separation toolkit',
long_description=long_description,
long_description_content_type="text/markdown",
license='MIT',
python_requires='>=3.6',
install_requires=['pyyaml',
'numpy',
'scipy',
'pandas',
'soundfile',
'torch',
'pytorch-lightning>=0.7.5,<0.8',
'torch_optimizer',
'pb_bss_eval',
'torch_stoi',
],
extras_require={
'visualize': ['seaborn'],
'tests': ['pytest'],
},
entry_points={
'console_scripts': ['asteroid-upload=asteroid.scripts.asteroid_cli:upload'],
},
packages=find_packages(),
include_package_data=True,
classifiers=[
'Development Status :: 4 - Beta',
"Programming Language :: Python :: 3",
'Programming Language :: Python :: 3.6',
'Programming Language :: Python :: 3.7',
'Programming Language :: Python :: 3.8',
"License :: OSI Approved :: MIT License",
"Operating System :: OS Independent",
],
)