Suggest the right module

Author: rockerBOO

library/custom_train_functions.py

@@ -588,7 +588,7 @@ def __init__(self, wavelet='db4', level=3, transform="dwt", loss_fn=torch.nn.fun
         self.hl_weight2 = 0.01
         self.hh_weight2 = 0.05
 
-        assert pywt.wavedec2 is not None, "PyWavelet module not available. Please install `pip install PyWavelet`"
+        assert pywt.wavedec2 is not None, "PyWavelets module not available. Please install `pip install PyWavelets`"
         # Create GPU filters from wavelet
         wav = pywt.Wavelet(wavelet)
         self.register_buffer('dec_lo', torch.Tensor(wav.dec_lo).to(device))