Fixed import of importlib.metadata

This commit is contained in:
romi1502
2021-03-01 11:54:47 +01:00
parent 56bb993a39
commit 4017216423
3 changed files with 26 additions and 1 deletions

View File

@@ -56,6 +56,7 @@ tensorflow = "2.3.0"
pandas = "1.1.2" pandas = "1.1.2"
numpy = "<1.19.0,>=1.16.0" numpy = "<1.19.0,>=1.16.0"
importlib-resources = {version = "^4.1.1", python = "<3.7"} importlib-resources = {version = "^4.1.1", python = "<3.7"}
importlib-metadata = {version = "^3.0.0", python = "<3.8"}
[tool.poetry.dev-dependencies] [tool.poetry.dev-dependencies]
pytest = "^6.2.1" pytest = "^6.2.1"

View File

@@ -130,7 +130,10 @@ VerboseOption: OptionInfo = Option(False, "--verbose", help="Enable verbose logs
def version_callback(value: bool): def version_callback(value: bool):
if value: if value:
from importlib.metadata import version try:
from importlib.metadata import version
except ImportError:
from importlib_metadata import version
echo(f"Spleeter Version: {version('spleeter')}") echo(f"Spleeter Version: {version('spleeter')}")
raise Exit() raise Exit()

21
tests/test_command.py Normal file
View File

@@ -0,0 +1,21 @@
#!/usr/bin/env python
# coding: utf8
""" Unit testing for Separator class. """
__email__ = 'research@deezer.com'
__author__ = 'Deezer Research'
__license__ = 'MIT License'
from spleeter.__main__ import spleeter
from typer.testing import CliRunner
def test_version():
runner = CliRunner()
# execute spleeter version command
result = runner.invoke(spleeter, [
'--version',
])