fix: add musdb and museval check

spleeter/commands/evaluate.py

@@ -13,6 +13,7 @@
         --mus_dir /path/to/musdb dataset
 """
 
+import sys
 import json
 
 from argparse import Namespace
@@ -21,8 +22,6 @@ from glob import glob
 from os.path import join, exists
 
 # pylint: disable=import-error
-import musdb
-import museval
 import numpy as np
 import pandas as pd
 # pylint: enable=import-error
@@ -30,6 +29,15 @@ import pandas as pd
 from .separate import entrypoint as separate_entrypoint
 from ..utils.logging import get_logger
 
+try:
+    import musdb
+    import museval
+except ImportError:
+    logger = get_logger()
+    logger.error('Extra dependencies musdb and museval not found')
+    logger.error('Please install musdb and museval first, abort')
+    sys.exit(1)
+
 __email__ = 'research@deezer.com'
 __author__ = 'Deezer Research'
 __license__ = 'MIT License'

spleeter/model/functions/unet.py

@@ -3,10 +3,9 @@
 
 """
 This module contains building functions for U-net source separation source
-separation models.
-Each instrument is modeled by a single U-net convolutional/deconvolutional
-network that take a mix spectrogram as input and the estimated sound spectrogram
-as output.
+separation models. Each instrument is modeled by a single U-netconvolutional
+/ deconvolutional network that take a mix spectrogram as input and the
+estimated sound spectrogram as output.
 """
 
 from functools import partial