Adding exceptions for hl and cfs

Signed-off-by: Gerardo Roa Dabike <gerardo.roa@gmail.com>
claritychallenge · Nov 15, 2022 · 00d29ca · 00d29ca
1 parent bbc5f9d
commit 00d29ca
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/clarity/enhancer/nalr.py b/clarity/enhancer/nalr.py
@@ -131,8 +131,12 @@ def __init__(self, nfir: int, fs: int):
         self.delay[nfir // 2] = 1.0
 
     def hl_interp(self, hl: np.ndarray, cfs: np.ndarray):
-        assert len(hl) == len(cfs), "Hearing losses and center frequencies don't match!"
-        hl_interpf = scipy.interpolate.interp1d(cfs, hl)
+        try:
+            hl_interpf = scipy.interpolate.interp1d(cfs, hl)
+        except ValueError:
+            raise ValueError(
+                "Hearing losses (hl) and center frequencies (cfs) don't match!"
+            )
         return hl_interpf(self.aud)
 
     def build(