Add comment to code

audeering · Jan 29, 2024 · 8b725bc · 8b725bc
1 parent cc799ce
commit 8b725bc
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/auglib/core/transform.py b/auglib/core/transform.py
@@ -3261,10 +3261,14 @@ def _pink_noise(self, samples: int) -> np.ndarray:
         def psd(f):
             return 1 / np.where(f == 0, float("inf"), np.sqrt(f))
 
-        white_noise = np.fft.rfft(np.random.randn(samples + 1))
+        # Add extra sample
+        # to ensure correct length for odd samples
+        length = samples + 1
+
+        white_noise = np.fft.rfft(np.random.randn(length))
 
         # Normalized pink noise shape
-        pink_shape = psd(np.fft.rfftfreq(samples + 1))
+        pink_shape = psd(np.fft.rfftfreq(length))
         pink_shape = pink_shape / np.sqrt(np.mean(pink_shape**2))
 
         white_noise_shaped = white_noise * pink_shape