From e0bfab42f763e3c76676c5c5085c2b92a7e6e7db Mon Sep 17 00:00:00 2001
From: GeorgiosEfstathiadis
 <54844705+GeorgeEfstathiadis@users.noreply.github.com>
Date: Thu, 22 Feb 2024 15:28:36 -0500
Subject: [PATCH] change coefs dtype to account for large squares overflowing
 (#243)

---
 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)