PySDKit 0.4.21__tar.gz → 0.4.22__tar.gz

{pysdkit-0.4.21 → pysdkit-0.4.22}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: PySDKit
-Version: 0.4.21
+Version: 0.4.22
 Summary: A Python library for signal decomposition algorithms with a unified interface.
 Home-page: https://github.com/wwhenxuan/PySDKit
 Author: whenxuan, changewam, josefinez, Yuan Feng
@@ -11,13 +11,11 @@ Classifier: Intended Audience :: Science/Research
 Classifier: Topic :: Scientific/Engineering :: Mathematics
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Classifier: License :: OSI Approved :: MIT License
-Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.6
-Classifier: Programming Language :: Python :: 3.7
 Classifier: Programming Language :: Python :: 3.8
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
-Requires-Python: >=3.6
+Classifier: Programming Language :: Python :: 3.11
+Requires-Python: >=3.8
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: numpy>=1.24.3

{pysdkit-0.4.21 → pysdkit-0.4.22}/PySDKit.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: PySDKit
-Version: 0.4.21
+Version: 0.4.22
 Summary: A Python library for signal decomposition algorithms with a unified interface.
 Home-page: https://github.com/wwhenxuan/PySDKit
 Author: whenxuan, changewam, josefinez, Yuan Feng
@@ -11,13 +11,11 @@ Classifier: Intended Audience :: Science/Research
 Classifier: Topic :: Scientific/Engineering :: Mathematics
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Classifier: License :: OSI Approved :: MIT License
-Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.6
-Classifier: Programming Language :: Python :: 3.7
 Classifier: Programming Language :: Python :: 3.8
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
-Requires-Python: >=3.6
+Classifier: Programming Language :: Python :: 3.11
+Requires-Python: >=3.8
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: numpy>=1.24.3

{pysdkit-0.4.21 → pysdkit-0.4.22}/pysdkit/__init__.py

@@ -2,7 +2,7 @@
 A Python library for signal decomposition algorithms.
 """
 
-__version__ = "0.4.21"
+__version__ = "0.4.22"
 
 # Empirical Mode Decomposition
 from ._emd import EMD

{pysdkit-0.4.21 → pysdkit-0.4.22}/pysdkit/_emd/efd.py

@@ -6,7 +6,7 @@ Created on 2025/04/16 22:51:55
 """
 import numpy as np
 
-from typing import Optional, Tuple
+from typing import Optional, Tuple, Union
 
 from pysdkit.utils import fft, ifft, fmirror
 
@@ -33,7 +33,7 @@ class EFD(object):
 
     def __call__(
         self, signal: np.ndarray, return_all: Optional[bool] = False
-    ) -> np.ndarray | Tuple[np.ndarray, float]:
+    ) -> Union[np.ndarray, Tuple[np.ndarray, float]]:
         """allow instances to be called like functions"""
         return self.fit_transform(signal=signal, return_all=return_all)
 
@@ -43,7 +43,7 @@ class EFD(object):
 
     def fit_transform(
         self, signal: np.ndarray, return_all: Optional[bool] = False
-    ) -> np.ndarray | Tuple[np.ndarray, float]:
+    ) -> Union[np.ndarray, Tuple[np.ndarray, float]]:
         """
         Signal decomposition using EFD algorithm
 

{pysdkit-0.4.21 → pysdkit-0.4.22}/pysdkit/_emd/emd.py

@@ -9,7 +9,7 @@ Code taken from https://github.com/laszukdawid/PyEMD/blob/master/PyEMD/EMD.py
 import numpy as np
 from numpy import ndarray
 from scipy.interpolate import interp1d
-from typing import Optional, Tuple
+from typing import Optional, Tuple, Union
 
 from pysdkit._emd import akima, cubic, pchip, cubic_hermite, cubic_spline_3pts
 from pysdkit._emd import find_extrema_parabol, find_extrema_simple
@@ -231,7 +231,7 @@ class EMD(object):
 
     def extract_max_min_spline(
         self, time: np.ndarray, signal: np.ndarray
-    ) -> list[int] | tuple[ndarray, ndarray, ndarray, ndarray]:
+    ) -> Union[list[int], tuple[ndarray, ndarray, ndarray, ndarray]]:
         """
         Extracts top and bottom envelopes based on the signal,
         which are constructed based on maxima and minima, respectively.

{pysdkit-0.4.21 → pysdkit-0.4.22}/pysdkit/_emd/hht/hht.py

@@ -7,7 +7,7 @@ Created on 2025/02/06 10:29:05
 import numpy as np
 from matplotlib import pyplot as plt
 
-from typing import Optional, Tuple, Any
+from typing import Optional, Tuple, Any, Union
 
 from pysdkit.plot import plot_IMFs, plot_HilbertSpectrum
 from pysdkit.utils import (
@@ -47,7 +47,7 @@ class HHT(object):
         signal: np.ndarray,
         fs: Optional[float] = None,
         return_all: Optional[bool] = False,
-    ) -> np.ndarray | Tuple[np.ndarray, np.ndarray, np.ndarray]:
+    ) -> Union[np.ndarray, Tuple[np.ndarray, np.ndarray, np.ndarray]]:
         """
         Perform Hilbert-Huang transform on the signal `x`, and return the amplitude and
         instantaneous frequency function of each intrinsic mode.
@@ -64,7 +64,7 @@ class HHT(object):
         """Get the full name and abbreviation of the algorithm"""
         return "Hilbert-Huang Transform (HHT)"
 
-    def _get_emd(self) -> EMD | REMD | EEMD | CEEMDAN:
+    def _get_emd(self) -> Union[EMD, REMD, EEMD, CEEMDAN]:
         """选择对应的经验模态分解算法"""
         if self.algorithm == "EMD":
             return EMD(max_imfs=self.max_imfs)
@@ -107,7 +107,7 @@ class HHT(object):
         signal: np.ndarray,
         fs: Optional[float] = None,
         return_all: Optional[bool] = False,
-    ) -> np.ndarray | Tuple[np.ndarray, np.ndarray, np.ndarray]:
+    ) -> Union[np.ndarray, Tuple[np.ndarray, np.ndarray, np.ndarray]]:
         """
         Perform Hilbert-Huang transform on the signal `x`, and return the amplitude and
         instantaneous frequency function of each intrinsic mode.
@@ -207,7 +207,7 @@ class HHT(object):
         freq_lim: Optional[tuple[float, float]] = None,
         freq_res: Optional[float] = None,
         time_scale: int = 1,
-    ) -> tuple[list[Any] | Any, list[Any]] | list[Any] | Any:
+    ) -> Union[tuple[Union[list[Any], Any], list[Any]], list[Any], Any]:
         """
         Obtaining and visualizing the Hilbert spectrum.
 

{pysdkit-0.4.21 → pysdkit-0.4.22}/pysdkit/_emd/memd.py

@@ -12,7 +12,7 @@ from sys import exit
 
 from numpy import ndarray, dtype
 from scipy.interpolate import CubicSpline
-from typing import Optional, Tuple, Any
+from typing import Optional, Tuple, Any, Union
 
 
 class MEMD(object):
@@ -121,7 +121,6 @@ class MEMD(object):
         dir_vec = np.zeros(shape=(N_dim, 1))
 
         for it in range(0, self.n_dir):
-
             if N_dim != 3:
                 # Multivariate signal (for N_dim ~=3) with hammersley sequence
                 # Linear normalisation of hammersley sequence in the range of -1.00 - 1.00
@@ -495,7 +494,6 @@ def nth_prime(n: int) -> list:
 
     # Iterate through natural numbers starting from 3
     for i in range(3, 104745):  # 104745 is an arbitrary upper limit for demonstration
-
         # Check if the current number is prime
         if is_prime(i):
             # Append the prime number to the list
@@ -670,16 +668,16 @@ def boundary_conditions(
     x: np.ndarray,
     z: np.ndarray,
     nbsym: int,
-) -> (
-    Tuple[None, None, None, None, int]
-    | Tuple[
+) -> Union[
+    Tuple[None, None, None, None, int],
+    Tuple[
         ndarray[Any, dtype[Any]],
         ndarray[Any, dtype[Any]],
         ndarray[Any, dtype[Any]],
         ndarray[Any, dtype[Any]],
         int,
-    ]
-):
+    ],
+]:
     """
     Handle boundary conditions for signal processing by extending the signal symmetrically.
 

{pysdkit-0.4.21 → pysdkit-0.4.22}/pysdkit/_emd/remd.py

@@ -124,13 +124,21 @@ class REMD(object):
         """
         # Get the position of the extreme point by calling the function to find the extreme point
         if self.extrema_detection == "parabol":
-            local_max_pos, local_max_val, local_min_pos, local_min_val, indzer = (
-                find_extrema_parabol(time=time, signal=signal)
-            )
+            (
+                local_max_pos,
+                local_max_val,
+                local_min_pos,
+                local_min_val,
+                indzer,
+            ) = find_extrema_parabol(time=time, signal=signal)
         elif self.extrema_detection == "simple":
-            local_max_pos, local_max_val, local_min_pos, local_min_val, indzer = (
-                find_extrema_simple(time=time, signal=signal)
-            )
+            (
+                local_max_pos,
+                local_max_val,
+                local_min_pos,
+                local_min_val,
+                indzer,
+            ) = find_extrema_simple(time=time, signal=signal)
         else:
             raise ValueError("`find_extrema` must be 'parabol' or 'simple'")
 
@@ -241,7 +249,6 @@ class REMD(object):
 
         # in case symmetrized parts do not extend enough
         if tl_min[0] > time[0] or tl_max[0] > time[0]:
-
             if lsym == indmax[0]:
                 lmax = np.fliplr(indmax[: min(end_max, nbsym)])
 
@@ -254,7 +261,6 @@ class REMD(object):
             lsym = 0
 
         if tr_min[-1] < time[xlen] or tr_max[-1] < time[xlen]:
-
             if rsym == indmax[-1]:
                 rmax = np.fliplr(indmax[max(end_max - nbsym + 1, 0) :])
 

{pysdkit-0.4.21 → pysdkit-0.4.22}/pysdkit/_emd/tvf_emd.py

@@ -14,7 +14,7 @@ from scipy.sparse.linalg import spsolve
 from pysdkit.utils import inst_freq_local
 from pysdkit.utils import divide2exp
 
-from typing import Optional, Tuple, Dict, Any, Iterable
+from typing import Optional, Tuple, Dict, Any, Iterable, Union
 
 warnings.filterwarnings("ignore")
 
@@ -145,7 +145,7 @@ class TVF_EMD(object):
         bis_freq: np.ndarray,
         ind_remove_pad: np.ndarray,
         num_padding: int,
-    ) -> None | int | float | complex | ndarray | Iterable:
+    ) -> Union[None, int, float, complex, ndarray, Iterable]:
         """Remove unstable parts or 'noise' from the signal based on the extrema points of the input signal and certain rules, and smooth the signal through interpolation"""
         org_bis_freq = bis_freq.copy()
         flag_intermitt = 0
@@ -528,7 +528,7 @@ def check_knots(
     return x, y, knots
 
 
-def spline_base(breaks: np.ndarray, n: int) -> Dict[str, int | Any | Any]:
+def spline_base(breaks: np.ndarray, n: int) -> Dict[str, Union[int, Any, Any]]:
     """
     Generates B-spline base of order `n` for knots `breaks`
     This function comes from https://github.com/stfbnc/pytvfemd/tree/master

{pysdkit-0.4.21 → pysdkit-0.4.22}/pysdkit/_emd2d/emd2d.py

@@ -300,7 +300,6 @@ class EMD2D(object):
 
                 # Make sure there are enough extreme points
                 if len(min_peaks[0]) > 4 and len(max_peaks[0]) > 4:
-
                     # Record the last input data
                     imf_old = imf.copy()
 

{pysdkit-0.4.21 → pysdkit-0.4.22}/pysdkit/_esmd/esmd.py

@@ -8,6 +8,5 @@ import numpy as np
 
 
 class ESMD(object):
-
     def __init__(self):
         pass

{pysdkit-0.4.21 → pysdkit-0.4.22}/pysdkit/_ewt/ewt.py

@@ -13,7 +13,7 @@ Original Code from: https://github.com/vrcarva/ewtpy
 """
 import numpy as np
 from scipy.ndimage.filters import gaussian_filter
-from typing import Optional, Tuple
+from typing import Optional, Tuple, Union
 
 
 class EWT(object):
@@ -60,7 +60,7 @@ class EWT(object):
         signal: np.ndarray,
         N: Optional[int] = None,
         return_all: Optional[bool] = False,
-    ) -> Tuple[np.ndarray, np.ndarray, np.ndarray] | np.ndarray:
+    ) -> Union[Tuple[np.ndarray, np.ndarray, np.ndarray], np.ndarray]:
         """allow instances to be called like functions"""
         return self.fit_transform(signal=signal, N=N, return_all=return_all)
 
@@ -80,7 +80,7 @@ class EWT(object):
         signal: np.ndarray,
         N: Optional[int] = None,
         return_all: Optional[bool] = False,
-    ) -> Tuple[np.ndarray, np.ndarray, np.ndarray] | np.ndarray:
+    ) -> Union[Tuple[np.ndarray, np.ndarray, np.ndarray], np.ndarray]:
         """
         Perform Empirical Wavelet Transform on the input signal.
 
@@ -251,9 +251,9 @@ def EWT_Boundaries_Detect(
 
     elif reg == "gaussian":
         regFilter = np.zeros(lengthFilter)
-        regFilter[regFilter.size // 2] = (
-            1  # Ensure center is set for Gaussian filtering
-        )
+        regFilter[
+            regFilter.size // 2
+        ] = 1  # Ensure center is set for Gaussian filtering
         presig = np.convolve(
             ff, gaussian_filter(regFilter, sigmaFilter), mode="same"
         )  # Gaussian filter

{pysdkit-0.4.21 → pysdkit-0.4.22}/pysdkit/_faemd/extrema.py

@@ -7,15 +7,14 @@ Created on 2025/02/24 13:20:36
 """
 import numpy as np
 
-from typing import Tuple
+from typing import Tuple, Union
 
 
 def extrema(
     x: np.ndarray,
-) -> (
-    Tuple[np.ndarray, np.ndarray, np.ndarray, np.ndarray]
-    | Tuple[None, None, None, None]
-):
+) -> Union[
+    Tuple[np.ndarray, np.ndarray, np.ndarray, np.ndarray], Tuple[None, None, None, None]
+]:
     """
     Gets the global extrema points from a time series
 

{pysdkit-0.4.21 → pysdkit-0.4.22}/pysdkit/_faemd/faemd.py

@@ -60,7 +60,7 @@ class FAEMD(object):
         signal: np.ndarray,
         return_all: bool = False,
         max_imfs: Optional[int] = None,
-    ) -> Tuple[np.ndarray, np.ndarray, np.ndarray, np.ndarray] | np.ndarray:
+    ) -> Union[Tuple[np.ndarray, np.ndarray, np.ndarray, np.ndarray], np.ndarray]:
         """allow instances to be called like functions"""
         return self.fit_transform(
             signal=signal, return_all=return_all, max_imfs=max_imfs
@@ -195,7 +195,7 @@ class FAEMD(object):
         signal: np.ndarray,
         return_all: bool = False,
         max_imfs: Optional[int] = None,
-    ) -> Tuple[np.ndarray, np.ndarray, np.ndarray, np.ndarray] | np.ndarray:
+    ) -> Union[Tuple[np.ndarray, np.ndarray, np.ndarray, np.ndarray], np.ndarray]:
         """
         Execute the signal decomposition algorithm
 
@@ -234,7 +234,6 @@ class FAEMD(object):
 
         # 开始进行信号的迭代分解
         while imf < max_imfs - 1:
-
             # Initialising intermediary IMFs
             H = Residue.copy()
 

{pysdkit-0.4.21 → pysdkit-0.4.22}/pysdkit/_hvd/hvd.py

@@ -6,7 +6,7 @@ Created on 2025/02/06 00:19:23
 """
 import numpy as np
 
-from typing import Optional, Tuple
+from typing import Optional, Tuple, Union
 
 from pysdkit.utils import differ
 from pysdkit.utils import fft, ifft
@@ -57,7 +57,7 @@ class HVD(object):
 
     def __call__(
         self, signal: np.ndarray, return_all: Optional[bool] = False
-    ) -> Tuple[np.ndarray, np.ndarray] | np.ndarray:
+    ) -> Union[Tuple[np.ndarray, np.ndarray], np.ndarray]:
         """allow instances to be called like functions"""
         return self.fit_transform(signal, return_all)
 
@@ -112,7 +112,7 @@ class HVD(object):
 
     def fit_transform(
         self, signal: np.ndarray, return_all: Optional[bool] = False
-    ) -> Tuple[np.ndarray, np.ndarray] | np.ndarray:
+    ) -> Union[Tuple[np.ndarray, np.ndarray], np.ndarray]:
         """Start executing the Hilbert Vibration Decomposition algorithm."""
 
         # Mirror the input signal to improve decomposition at the endpoints

{pysdkit-0.4.21 → pysdkit-0.4.22}/pysdkit/_jmd/jmd.py

@@ -8,7 +8,7 @@ import numpy as np
 from numpy import linalg
 import scipy.sparse as sp
 
-from typing import Optional, Tuple
+from typing import Optional, Tuple, Union
 
 from pysdkit.utils import fft, fftshift, ifft, ifftshift
 
@@ -69,7 +69,7 @@ class JMD(object):
 
     def __call__(
         self, signal: np.ndarray, return_all: bool = True
-    ) -> Tuple[np.ndarray, np.ndarray, np.ndarray] | np.ndarray:
+    ) -> Union[Tuple[np.ndarray, np.ndarray, np.ndarray], np.ndarray]:
         """allow instances to be called like functions"""
         return self.fit_transform(signal=signal, return_all=return_all)
 
@@ -77,7 +77,9 @@ class JMD(object):
         """Get the full name and abbreviation of the algorithm"""
         return "Jump Plus AM-FM Mode Decomposition (JMD)"
 
-    def jump_step(self, freqs: np.ndarray, T: int) -> Tuple[
+    def jump_step(
+        self, freqs: np.ndarray, T: int
+    ) -> Tuple[
         float,
         np.ndarray,
         np.ndarray,
@@ -194,7 +196,7 @@ class JMD(object):
 
     def fit_transform(
         self, signal: np.ndarray, return_all: Optional[bool] = False
-    ) -> Tuple[np.ndarray, np.ndarray, np.ndarray] | np.ndarray:
+    ) -> Union[Tuple[np.ndarray, np.ndarray, np.ndarray], np.ndarray]:
         """
         Signal decomposition using Jump Plus AM-FM Mode Decomposition algorithm
 

{pysdkit-0.4.21 → pysdkit-0.4.22}/pysdkit/_lmd/rlmd.py

@@ -6,7 +6,7 @@ Created on 2025/01/31 23:33:43
 """
 import numpy as np
 
-from typing import Optional, Tuple
+from typing import Optional, Union, Tuple
 
 
 class RLMD(object):
@@ -86,7 +86,12 @@ class RLMD(object):
     def _initial_inputs(
         self, signal: np.ndarray
     ) -> Tuple[
-        np.ndarray | float, np.ndarray, np.ndarray, np.ndarray, np.ndarray, np.ndarray
+        Union[np.ndarray, float],
+        np.ndarray,
+        np.ndarray,
+        np.ndarray,
+        np.ndarray,
+        np.ndarray,
     ]:
         """"""
         # get the length of inputs signal
@@ -108,7 +113,7 @@ class RLMD(object):
 
         return signal_energy, ams, fms, pfs, iterNum, fvs
 
-    def _stop_lmd(self, signal: np.ndarray, signal_energy: np.ndarray | float) -> bool:
+    def _stop_lmd(self, signal: np.ndarray, signal_energy: Union[np.ndarray, float]) -> bool:
         """
         Stopping criterion of LMD algorithm.
         Check if there are enough (3) extrema to continue the decomposition
@@ -135,7 +140,7 @@ class RLMD(object):
 
     def _lmd_mean_amp(
         self, x: np.ndarray
-    ) -> Tuple[np.ndarray, np.ndarray, np.ndarray | int]:
+    ) -> Tuple[np.ndarray, np.ndarray, Union[np.ndarray, int]]:
         """
         Compute mean function and amplitude function of x in LMD
 
@@ -232,7 +237,7 @@ class RLMD(object):
 
     def _get_best_span(
         self, ind_extr: np.ndarray, x: np.ndarray
-    ) -> Tuple[np.ndarray | int, int]:
+    ) -> Tuple[Union[np.ndarray, int], int]:
         """
         Moving average span selection.
         Return the best span of moving average
@@ -315,7 +320,7 @@ class RLMD(object):
 
     def fit_transform(
         self, signal: np.ndarray, return_all: Optional[bool] = False
-    ) -> np.ndarray | Tuple[np.ndarray, np.ndarray, np.ndarray]:
+    ) -> Union[np.ndarray, Tuple[np.ndarray, np.ndarray, np.ndarray]]:
         """
         This funcion perform the local mean decompose (LMD) on the input signal, and return the product function (pfs),
         and their corresponding instantaneous amplititde(ams) and frequency modulation signal(fms).
@@ -338,7 +343,6 @@ class RLMD(object):
         while i < self.max_imfs and not self._stop_lmd(
             signal=xs, signal_energy=signal_energy
         ):
-
             # initialize variables used in PF sifting loop
             a_i = np.ones(nx)
             s_j = np.zeros(shape=(self.max_iter, nx))
@@ -492,7 +496,7 @@ def extr(x: np.ndarray) -> Tuple[np.ndarray, np.ndarray, np.ndarray]:
 
 
 def extend(
-    x: np.ndarray, indmin: np.ndarray, indmax: np.ndarray, extd_r: np.ndarray | float
+    x: np.ndarray, indmin: np.ndarray, indmax: np.ndarray, extd_r: Union[np.ndarray, float]
 ) -> Tuple[np.ndarray, np.ndarray, np.ndarray, np.ndarray]:
     """
     Python implementation of the MATLAB `extend` routine from the EMD toolbox.
@@ -664,7 +668,7 @@ def rms(
     *,
     keepdims: bool = False,
     omitnan: bool = False,
-) -> np.ndarray | float:
+) -> Union[np.ndarray, float]:
     """
     The Python version of rms function in Matlab.
 
@@ -690,7 +694,7 @@ def kurtosis(
     axis: Optional[int] = 0,
     bias: Optional[bool] = False,
     nan_policy: Optional[str] = "propagate",
-) -> np.ndarray | float:
+) -> Union[np.ndarray, float]:
     """
     Reproduce MATLAB kurtosis(x) behavior (global formula, flag=0)
 

{pysdkit-0.4.21 → pysdkit-0.4.22}/pysdkit/_vmd/base.py

@@ -73,9 +73,9 @@ class Base(object):
         fMirr[:, 0 : T // 2] = np.flip(
             ts[:, 0 : T // 2], axis=1
         )  # Mirror the first T//2 elements at the beginning
-        fMirr[:, T // 2 : 3 * T // 2] = (
-            ts  # Include the original time series in the middle
-        )
+        fMirr[
+            :, T // 2 : 3 * T // 2
+        ] = ts  # Include the original time series in the middle
         fMirr[:, 3 * T // 2 : 2 * T] = np.flip(
             ts[:, T // 2 :], axis=1
         )  # Mirror the last T//2 elements at the end

{pysdkit-0.4.21 → pysdkit-0.4.22}/pysdkit/_vmd/mvmd.py

@@ -5,7 +5,7 @@ Created on 2024/6/1 18:44
 @email: wwhenxuan@gmail.com
 """
 import numpy as np
-from typing import Optional, Tuple
+from typing import Optional, Tuple, Union
 from .base import Base
 
 
@@ -101,7 +101,7 @@ class MVMD(Base):
 
     def fit_transform(
         self, signal: np.ndarray, return_all: bool = False
-    ) -> Tuple[np.ndarray, np.ndarray, np.ndarray] | np.ndarray:
+    ) -> Union[Tuple[np.ndarray, np.ndarray, np.ndarray], np.ndarray]:
         """
         Multivariate signal decomposition using MVMD algorithm
 
@@ -179,7 +179,6 @@ class MVMD(Base):
                 ) / np.sum(np.square(np.abs(u_hat_plus[n, T // 2 : T, k - 1, :])))
 
             for k in range(2, self.K + 1):
-
                 # update mode accumulator
                 sum_uk = (
                     u_hat_plus[n, :, k - 2, :] + sum_uk - u_hat_plus[n - 1, :, k - 1, :]

{pysdkit-0.4.21 → pysdkit-0.4.22}/pysdkit/_vmd/svmd.py

@@ -211,7 +211,8 @@ class SVMD(Base):
                     # update u_L
                     u_hat_L[n + 1, :] = (
                         f_hat_onesided
-                        + ((Alpha**2) * (omega_freqs - omega_L[n]) ** 4) * u_hat_L[n, :]
+                        + ((Alpha**2) * (omega_freqs - omega_L[n]) ** 4)
+                        * u_hat_L[n, :]
                         + lambda_vector[n, :] / 2
                     ) / (
                         1
@@ -304,7 +305,6 @@ class SVMD(Base):
             if self.init_omega > 0:
                 ii = 0
                 while ii < 1 and n2 < 300:
-
                     omega_L = np.sort(
                         np.exp(
                             np.log(fs) + (np.log(0.5) - np.log(fs)) * self.rng.rand()

{pysdkit-0.4.21 → pysdkit-0.4.22}/pysdkit/_vmd/vmd_c.py

@@ -5,7 +5,7 @@ Created on Sat Mar 4 11:59:21 2024
 @email: wwhenxuan@gmail.com
 """
 import numpy as np
-from typing import Optional, List, Tuple
+from typing import Optional, List, Tuple, Union
 
 from .base import Base
 from pysdkit.plot import plot_IMFs
@@ -149,7 +149,7 @@ class VMD(Base):
 
     def fit_transform(
         self, signal: np.ndarray, return_all: bool = False
-    ) -> Tuple[np.ndarray, np.ndarray, np.ndarray] | np.ndarray:
+    ) -> Union[Tuple[np.ndarray, np.ndarray, np.ndarray], np.ndarray]:
         """
         Signal decomposition using VMD algorithm
 

{pysdkit-0.4.21 → pysdkit-0.4.22}/pysdkit/_vmd/vme.py

@@ -6,7 +6,7 @@ Created on 2025/04/02 22:44:18
 """
 import numpy as np
 
-from typing import Optional, Tuple
+from typing import Optional, Union, Tuple
 
 from pysdkit.utils import fft, fftshift, ifft, ifftshift
 
@@ -63,7 +63,7 @@ class VME(object):
 
     def fit_transform(
         self, signal: np.ndarray, return_all: Optional[bool] = False
-    ) -> np.ndarray | Tuple[np.ndarray, np.ndarray, np.ndarray]:
+    ) -> Union[np.ndarray, Tuple[np.ndarray, np.ndarray, np.ndarray]]:
         """
         Signal decomposition using VME algorithm