From 9a10289ce5d7af8d2b5620d1cdf0536fd091cfe5 Mon Sep 17 00:00:00 2001 From: Amit Moscovich Date: Mon, 5 Aug 2024 17:20:39 +0300 Subject: [PATCH 1/2] Update wemd.py Mean-subtract the input array to wemd_embed --- src/aspire/operators/wemd.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/aspire/operators/wemd.py b/src/aspire/operators/wemd.py index 45db7203bb..8b1bfa72ce 100644 --- a/src/aspire/operators/wemd.py +++ b/src/aspire/operators/wemd.py @@ -46,7 +46,7 @@ def wemd_embed(arr, wavelet="coif3", level=None): message="Level value of .* is too high:" " all coefficients will experience boundary effects.", ) - arrdwt = pywt.wavedecn(arr, wavelet, mode="zero", level=level) + arrdwt = pywt.wavedecn(arr-arr.mean(), wavelet, mode="zero", level=level) detail_coefs = arrdwt[1:] assert len(detail_coefs) == level From db778fdde9bbfd9202ef34cdc5fb95833fa13a02 Mon Sep 17 00:00:00 2001 From: Garrett Wright Date: Mon, 5 Aug 2024 10:36:01 -0400 Subject: [PATCH 2/2] minor black style updates --- src/aspire/operators/wemd.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/aspire/operators/wemd.py b/src/aspire/operators/wemd.py index 8b1bfa72ce..09fd1db563 100644 --- a/src/aspire/operators/wemd.py +++ b/src/aspire/operators/wemd.py @@ -46,7 +46,7 @@ def wemd_embed(arr, wavelet="coif3", level=None): message="Level value of .* is too high:" " all coefficients will experience boundary effects.", ) - arrdwt = pywt.wavedecn(arr-arr.mean(), wavelet, mode="zero", level=level) + arrdwt = pywt.wavedecn(arr - arr.mean(), wavelet, mode="zero", level=level) detail_coefs = arrdwt[1:] assert len(detail_coefs) == level