The first implementation was for images thats correct
added implementation for DWT1D and IDWT1D that may help
https://github.com/Timorleiderman/tensorflow-wavelets/blob/642335ad5473531fb06ca11b921f9b8c99a1b144/src/tensorflow_wavelets/Layers/DWT.py#L201C7-L201C12