Merge pull request #25 from roxyboy/args

_morlet takes **Kwargs
roxyboy · Aug 16, 2023 · aea2b61 · aea2b61
2 parents 39bd09e + 0d796e0
commit aea2b61
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/xwavelet/wavelet.py b/xwavelet/wavelet.py
@@ -249,7 +249,7 @@ def wvlt_power_spectrum(
         x = da[da.dims[axis_num[-1]]] - N[-1] / 2.0 * delta_x[-1]
         if wtype == "morlet":
             # mother wavelet
-            wavelet, phi = _morlet(xo, ntheta, a, 1.0, y, x, dim)
+            wavelet, phi = _morlet(xo, ntheta, a, 1.0, y, x, **kwargs)
 
         Fdims = []
         chunks = dict()
@@ -362,7 +362,7 @@ def wvlt_cross_spectrum(
         x = da[da.dims[axis_num[-1]]] - N[-1] / 2.0 * delta_x[-1]
         if wtype == "morlet":
             # mother wavelet
-            wavelet, phi = _morlet(xo, ntheta, a, 1.0, y, x, dim)
+            wavelet, phi = _morlet(xo, ntheta, a, 1.0, y, x, **kwargs)
 
         Fdims = []
         chunks = dict()