From b4f7e28648716bc75c646bfa8d058e2b1f985a07 Mon Sep 17 00:00:00 2001 From: Georgios Efstathiadis Date: Thu, 22 Feb 2024 15:15:50 -0500 Subject: [PATCH] change coefs dtype to account for large squares overflowing --- forest/oak/base.py | 1 + 1 file changed, 1 insertion(+) diff --git a/forest/oak/base.py b/forest/oak/base.py index f21ef82a..f2751833 100644 --- a/forest/oak/base.py +++ b/forest/oak/base.py @@ -176,6 +176,7 @@ def compute_interpolate_cwt(tapered_bout: np.ndarray, fs: int = 10, out = ssq_cwt(tapered_bout[:-1], wavelet, fs=10) coefs = out[0] coefs = np.append(coefs, coefs[:, -1:], 1) + coefs = coefs.astype('complex128') # magnitude of cwt coefs = np.abs(coefs**2)