fix: allocate temporal value to eps

keonlee9420 · Apr 23, 2021 · c396a3a · c396a3a
1 parent 089b4a9
commit c396a3a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/data/noise_mixer.py b/data/noise_mixer.py
@@ -52,6 +52,7 @@ def audioread(path, tg_path=None, norm=True, start=0, stop=None):
 # Funtion to write audio    
 def audiowrite(data, fs, destpath, norm=False):
     if norm:
+        eps = 1e-6
         rms = (data ** 2).mean() ** 0.5
         scalar = 10 ** (-25 / 10) / (rms+eps)
         data = data * scalar

diff --git a/data/noise_mixer_refs.py b/data/noise_mixer_refs.py
@@ -52,6 +52,7 @@ def audioread(path, tg_path=None, norm=True, start=0, stop=None):
 # Funtion to write audio    
 def audiowrite(data, fs, destpath, norm=False):
     if norm:
+        eps = 1e-6
         rms = (data ** 2).mean() ** 0.5
         scalar = 10 ** (-25 / 10) / (rms+eps)
         data = data * scalar