pytorch_wavelets cannot find pywt

[alyashev]

pip install pytorch_wavelets

$ python -c "import pytorch_wavelets"
Traceback (most recent call last):
File "", line 1, in
File "/home/alyashev/python/envs/pi-wno/lib/python3.10/site-packages/pytorch_wavelets/init.py", line 22, in
from pytorch_wavelets.dtcwt.transform2d import DTCWTForward, DTCWTInverse
File "/home/alyashev/python/envs/pi-wno/lib/python3.10/site-packages/pytorch_wavelets/dtcwt/transform2d.py", line 7, in
from pytorch_wavelets.dtcwt.transform_funcs import FWD_J1, FWD_J2PLUS
File "/home/alyashev/python/envs/pi-wno/lib/python3.10/site-packages/pytorch_wavelets/dtcwt/transform_funcs.py", line 7, in
from pytorch_wavelets.dwt.lowlevel import int_to_mode
File "/home/alyashev/python/envs/pi-wno/lib/python3.10/site-packages/pytorch_wavelets/dwt/lowlevel.py", line 6, in
import pywt
ModuleNotFoundError: No module named 'pywt'