Merge pull request aliutkus#5 from mpariente/optimize_imports

Optimize imports to lighten dependencies

setup.py

@@ -12,16 +12,16 @@
         'numpy',
         'scipy',
         'tqdm',
-        'librosa',
+        'resampy',
         'pystoi',
         'museval',
         'gammatone @ git+https://github.com/detly/gammatone',
         'pypesq @ git+https://github.com/vBaiCai/python-pesq',
-        'srmrpy @ git+https://github.com/jfsantos/SRMRpy'
+        'srmrpy @ git+https://github.com/jfsantos/SRMRpy',
     ],
     extras_require={
-        'cpu': ['tensorflow==2.0.0'],
-        'gpu': ['tensorflow-gpu==2.0.0'],
+        'cpu': ['tensorflow==2.0.0', 'librosa'],
+        'gpu': ['tensorflow-gpu==2.0.0', 'librosa'],
     },
     include_package_data=True
 )

speechmetrics/absolute/mosnet/__init__.py

@@ -1,7 +1,5 @@
-import tensorflow as tf
-
-
 def load(window, hop=None):
+    import tensorflow as tf
     from .model import MOSNet
     tf.debugging.set_log_device_placement(False)
     # set memory growth

speechmetrics/absolute/srmr/__init__.py

@@ -1,4 +1,3 @@
-from .srmr import srmr
 from ... import Metric
 
 
@@ -11,6 +10,7 @@ def __init__(self, window, hop=None):
         self.absolute = True
 
     def test_window(self, audios, rate):
+        from .srmr import srmr
         return {'srmr': srmr(audios[0], self.fixed_rate, n_cochlear_filters=23,
                              low_freq=125, min_cf=4,
                              max_cf=128, fast=True, norm=False)[0]}