refactor: set musdb and museval as extras

This commit is contained in:
Félix Voituret
2019-11-06 23:58:58 +01:00
parent d622a01ad3
commit 30cd7c6435

View File

@@ -14,7 +14,7 @@ __license__ = 'MIT License'
# Default project values. # Default project values.
project_name = 'spleeter' project_name = 'spleeter'
project_version = '1.4.1' project_version = '1.4.2'
device_target = 'cpu' device_target = 'cpu'
tensorflow_dependency = 'tensorflow' tensorflow_dependency = 'tensorflow'
tensorflow_version = '1.14.0' tensorflow_version = '1.14.0'
@@ -64,8 +64,6 @@ setup(
include_package_data=True, include_package_data=True,
install_requires=[ install_requires=[
'importlib_resources ; python_version<"3.7"', 'importlib_resources ; python_version<"3.7"',
'musdb==0.3.1',
'museval==0.3.0',
'norbert==0.2.1', 'norbert==0.2.1',
'pandas==0.25.1', 'pandas==0.25.1',
'requests', 'requests',
@@ -73,6 +71,9 @@ setup(
'ffmpeg-python', 'ffmpeg-python',
'{}=={}'.format(tensorflow_dependency, tensorflow_version), '{}=={}'.format(tensorflow_dependency, tensorflow_version),
], ],
extras_require={
'evaluation': ['musdb==0.3.1', 'museval==0.3.0']
},
entry_points={ entry_points={
'console_scripts': ['spleeter=spleeter.__main__:entrypoint'] 'console_scripts': ['spleeter=spleeter.__main__:entrypoint']
}, },