Skip to content

Commit 1932885

Browse files
committed
Fix integer requirement for padding parameters n1 and n2. Return the parameters for post-processing purposes
1 parent 0b61ef7 commit 1932885

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

pymodalib/implementations/python/wavelet/wavelet_transform.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -309,7 +309,8 @@ def wavelet_transform(
309309
else:
310310
n1 = np.floor((NL - L) * coib1[0] / (coib1[0] + coib2[0]))
311311
n2 = np.ceil((NL - L) * coib1[0] / (coib1[0] + coib2[0]))
312-
312+
n1 = int(n1)
313+
n2 = int(n2)
313314
if padding == "predictive":
314315
pow = (-(L / fs - np.arange(1, L + 1) / fs)) / (wp.t2h - wp.t1h)
315316
w = 2 ** pow
@@ -498,6 +499,8 @@ def wavelet_transform(
498499
"cut_edges": cut_edges,
499500
"fs": fs,
500501
"padding": padding,
502+
"n1": n1,
503+
"n2": n2,
501504
"rel_tolerance": rel_tolerance,
502505
}
503506
return WT, freq, opt

0 commit comments

Comments
 (0)