fix lint.

Author: v0lta

src/ptwt/packets.py

@@ -6,9 +6,9 @@
 from itertools import product
 from typing import TYPE_CHECKING, Callable, Dict, List, Optional, Tuple, Union, cast
 
+import numpy as np
 import pywt
 import torch
-import numpy as np
 
 from ._util import Wavelet, _as_wavelet
 from .conv_transform import wavedec, waverec
@@ -23,21 +23,22 @@
     BaseDict = collections.UserDict
 
 
-def _wpfreq(fs: float, level: int) -> np.ndarray:
-    """Compute the frequencies for a fully decomposed single dimensional
-       packet tree. The packet transform linearly subdivides all frequencies
+def _wpfreq(fs: float, level: int) -> List[float]:
+    """Compute the frequencies for a fully decomposed 1d packet tree.
+
+       The packet transform linearly subdivides all frequencies
        from zero up to the Nyquist frequency.
 
     Args:
-        fs (float): The sampling frequency
-        level (int): The decomposition level
+        fs (float): The sampling frequency.
+        level (int): The decomposition level.
 
     Returns:
-        np.ndarray: The frequency bins of the packets in frequency order.
+        List[float]: The frequency bins of the packets in frequency order.
     """
-    n = list(range(int(np.power(2., level))))
-    freqs = (fs/2.)*(n/(np.power(2., level)))
-    return freqs
+    n = np.array(range(int(np.power(2.0, level))))
+    freqs = (fs / 2.0) * (n / (np.power(2.0, level)))
+    return list(freqs)
 
 
 class WaveletPacket(BaseDict):