set_snr: add missing default for axis

Haibit · Aug 16, 2019 · 999233c · 999233c
1 parent 2735ea5
commit 999233c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nt/speech_enhancement/noise/utils.py b/nt/speech_enhancement/noise/utils.py
@@ -35,7 +35,7 @@ def get_snr(X, N, *, axis=None, keepdims=False):
     return 10 * np.log10(power_X / power_N)
 
 
-def set_snr(X, N, snr, current_snr=None, *, axis, inplace=True):
+def set_snr(X, N, snr, current_snr=None, *, axis=None, inplace=True):
     """
     Set the SNR of two input images by rescaling the noise signal in place.