From a51817916abe2d304a4ecb9086ee4525c28864d4 Mon Sep 17 00:00:00 2001 From: fsmaibrgm Date: Fri, 7 Apr 2023 17:49:34 +0200 Subject: [PATCH] fix reshape --- src/razorback/prefilters.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/razorback/prefilters.py b/src/razorback/prefilters.py index b58287b..364de36 100644 --- a/src/razorback/prefilters.py +++ b/src/razorback/prefilters.py @@ -27,7 +27,7 @@ def value(self, e, b): size = self._size coeff_det = self.coeff_determination breshape = lambda arr: arr[:len(arr)-(len(arr) % size)].reshape( - (divmod(len(arr), size)[0], size) + (-1,) * (arr.ndim-1) + (divmod(len(arr), size)[0], size) + arr.shape[1:] ) cod = np.zeros(len(e)) queue = size + (len(e) % size)