some small formatting

Author: xserra

smstools/models/dftModel.py

@@ -6,7 +6,12 @@
 # - `dftSynth`: synthesis of a time-domain frame from magnitude/phase
 # - `dftModel`: convenience analysis+synthesis round-trip for one frame
 
-__all__ = ["dftAnal", "dftSynth", "dftModel"]
+
+__all__ = [
+    "dftAnal",
+    "dftSynth",
+    "dftModel",
+]
 
 
 import numpy as np
@@ -29,9 +34,13 @@ def _validate_window_fft_size(w: np.ndarray, N: int) -> None:
         ValueError: If N is not a power of 2 or w is larger than N.
     """
     if not UF.isPower2(N):
-        raise ValueError(f"FFT size (N={N}) is not a power of 2. Provided window size: {w.size}.")
+        raise ValueError(
+            f"FFT size (N={N}) is not a power of 2. Provided window size: {w.size}."
+        )
     if w.size > N:
-        raise ValueError(f"Window size (M={w.size}) is bigger than FFT size (N={N}).")
+        raise ValueError(
+            f"Window size (M={w.size}) is bigger than FFT size (N={N})."
+        )
 
 def _positive_spectrum_from_fft(X: np.ndarray, hN: int) -> np.ndarray:
     """
@@ -107,7 +116,9 @@ def dftSynth(mX: np.ndarray, pX: np.ndarray, M: int) -> np.ndarray:
     hN = mX.size
     N = (hN - 1) * 2
     if not UF.isPower2(N):
-        raise ValueError(f"size of mX ({mX.size}) is not (N/2)+1 for N={N}. Check input spectrum size.")
+        raise ValueError(
+            f"size of mX ({mX.size}) is not (N/2)+1 for N={N}. Check input spectrum size."
+        )
     hM1 = (M + 1) // 2
     hM2 = M // 2
     y = np.zeros(M)

smstools/models/harmonicModel.py

@@ -6,6 +6,7 @@
 # - `harmonicModel`: analysis+synthesis round-trip
 # - `harmonicModelAnal` / `harmonicModelSynth`: separated analysis/synthesis APIs
 
+
 import math
 
 import numpy as np
@@ -77,7 +78,7 @@ def f0Detection(
         ipfreq = fs * iploc / N  # convert locations to Hz
         f0t = UF.f0Twm(ipfreq, ipmag, f0et, minf0, maxf0, f0stable, fs=fs)  # find f0
         f0stable = f0t if f0t > 0 and abs(f0t - f0stable) < f0et else 0
-        f0candidate = f0t
+        # f0candidate = f0t  # unused variable
         f0 = np.append(f0, f0t)  # add f0 to output array
         pin += H  # advance sound pointer
     return f0
@@ -206,7 +207,9 @@ def harmonicModelAnal(
         # f0candidate = f0t  # unused variable
         f0.append(f0t)  # add f0 to output list
         # Harmonic detection
-        hfreq, hmag, hphase = harmonicDetection(ipfreq, ipmag, ipphase, f0t, nH, hfreqp, fs, harmDevSlope)
+        hfreq, hmag, hphase = harmonicDetection(
+            ipfreq, ipmag, ipphase, f0t, nH, hfreqp, fs, harmDevSlope
+        )
         xhfreq.append(hfreq)
         xhmag.append(hmag)
         xhphase.append(hphase)
@@ -256,7 +259,9 @@ def harmonicModel(
     # User can specify N for analysis
     H = 128
     Ns = 512
-    hfreq, hmag, hphase = harmonicModelAnal(x, fs, w, N, H, t, nH, minf0, maxf0, f0et)
+    hfreq, hmag, hphase = harmonicModelAnal(
+        x, fs, w, N, H, t, nH, minf0, maxf0, f0et
+    )
     # Synthesize using fixed synthesis parameters
     y = SM.sineModelSynth(hfreq, hmag, hphase, Ns, H, fs)
     # Ensure output length matches input