From 64422ff4a01c7b4e96ce0f3eed5e563017c8b4f4 Mon Sep 17 00:00:00 2001
From: rockerBOO <rockerboo@gmail.com>
Date: Tue, 8 Apr 2025 04:13:37 -0400
Subject: [PATCH] Suggest the right module

---
 library/custom_train_functions.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/library/custom_train_functions.py b/library/custom_train_functions.py
index 85ee1dea..514f1a0a 100644
--- a/library/custom_train_functions.py
+++ b/library/custom_train_functions.py
@@ -588,7 +588,7 @@ class WaveletLoss(torch.nn.Module):
         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))