Skip to content

Commit 9c0ff9d

Browse files
committed
FIX: raise error on attempt to transform along an axis with size 0
1 parent 13a4725 commit 9c0ff9d

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

pywt/_extensions/_dwt.pyx

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -90,7 +90,7 @@ cpdef dwt_axis(np.ndarray data, Wavelet wavelet, MODE mode, unsigned int axis=0)
9090

9191
input_shape = <size_t [:data.ndim]> <size_t *> data.shape
9292
output_shape = input_shape.copy()
93-
output_shape[axis] = common.dwt_buffer_length(data.shape[axis], wavelet.dec_len, mode)
93+
output_shape[axis] = dwt_coeff_len(data.shape[axis], wavelet.dec_len, mode)
9494

9595
cA = np.empty(output_shape, data.dtype)
9696
cD = np.empty(output_shape, data.dtype)

0 commit comments

Comments
 (0)