Check output contains only valid numbers

jurihock · Mar 31, 2024 · 97a61cc · 97a61cc
1 parent 58bcabf
commit 97a61cc
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/remucs/tuning.py b/src/remucs/tuning.py
@@ -113,4 +113,7 @@ def analyze(src: Path, opts: RemucsOptions) -> Tuple[NDArray, NDArray]:
     estimates = estimates[latency:latency+oldsize]
     weights   = weights[latency:latency+oldsize]
 
+    assert numpy.all(numpy.isfinite(estimates))
+    assert numpy.all(numpy.isfinite(weights))
+
     return estimates, weights