Skip to content

Commit 6a9d413

Browse files
committed
fix pywt.max_level
1 parent 1a6863b commit 6a9d413

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

src/ptwt/packets.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,7 @@ def transform(
7979
"""
8080
self.data = {}
8181
if maxlevel is None:
82-
maxlevel = pywt.dwt_maxlevel(data.shape[-1], self.wavelet.dec_len)
82+
maxlevel = pywt.dwt_max_level(data.shape[-1], self.wavelet.dec_len)
8383
self.maxlevel = maxlevel
8484
self._recursive_dwt(data, level=0, path="")
8585
return self
@@ -105,7 +105,7 @@ def reconstruct(self) -> None:
105105
>>> print(ptwp[""])
106106
"""
107107
if self.maxlevel is None:
108-
self.maxlevel = pywt.dwt_maxlevel(self[""].shape[-1], self.wavelet.dec_len)
108+
self.maxlevel = pywt.dwt_max_level(self[""].shape[-1], self.wavelet.dec_len)
109109

110110
for level in reversed(range(self.maxlevel)):
111111
for node in self.get_level(level):
@@ -268,7 +268,7 @@ def transform(
268268
"""
269269
self.data = {}
270270
if maxlevel is None:
271-
maxlevel = pywt.dwt_maxlevel(min(data.shape[-2:]), self.wavelet.dec_len)
271+
maxlevel = pywt.dwt_max_level(min(data.shape[-2:]), self.wavelet.dec_len)
272272
self.maxlevel = maxlevel
273273

274274
self._recursive_dwt2d(data, level=0, path="")
@@ -283,7 +283,7 @@ def reconstruct(self) -> None:
283283
a reconstruction from the leafs.
284284
"""
285285
if self.maxlevel is None:
286-
self.maxlevel = pywt.dwt_maxlevel(
286+
self.maxlevel = pywt.dwt_max_level(
287287
min(self[""].shape[-2:]), self.wavelet.dec_len
288288
)
289289

0 commit comments

Comments
 (0)