Change name of import reference in test to reflect name change.

sdatkinson · Dec 17, 2024 · c53ad7e · c53ad7e
1 parent e91d9d1
commit c53ad7e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_nam/test_train/test_lightning_module.py b/tests/test_nam/test_train/test_lightning_module.py
@@ -48,7 +48,7 @@ def mocked_loss(
             raise RuntimeError("Trigger fallback")
         return _torch.tensor(1.0)
 
-    mocker.patch("nam.train.lightning_module.multi_resolution_stft_loss", mocked_loss)
+    mocker.patch("nam.train.lightning_module._multi_resolution_stft_loss", mocked_loss)
 
     batch_size = 3
     sequence_length = 4096