change coefs dtype to account for large squares overflowing (#243)

onnela-lab · Feb 22, 2024 · e0bfab4 · e0bfab4
1 parent 02c4a6c
commit e0bfab4
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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)