From 457bf4c7c3228d7a6dab1f04e0e2a56b3cbc12bf Mon Sep 17 00:00:00 2001 From: Adam Thompson Date: Tue, 5 Oct 2021 08:30:08 -0400 Subject: [PATCH] Fix indexing error in CWT --- python/cusignal/wavelets/wavelets.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/cusignal/wavelets/wavelets.py b/python/cusignal/wavelets/wavelets.py index 8e0d1aa4..75cf10c3 100644 --- a/python/cusignal/wavelets/wavelets.py +++ b/python/cusignal/wavelets/wavelets.py @@ -307,6 +307,6 @@ def cwt(data, wavelet, widths): """ output = cp.empty([len(widths), len(data)]) for ind, width in enumerate(widths): - wavelet_data = wavelet(min(10 * width, len(data)), width) + wavelet_data = wavelet(min(10 * int(width), len(data)), int(width)) output[ind, :] = convolve(data, wavelet_data, mode="same") return output