Fix test.

LCAV · Jul 9, 2024 · 4d97665 · 4d97665
1 parent c54097a
commit 4d97665
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/test_algos.py b/test/test_algos.py
@@ -169,10 +169,10 @@ def test_trainable_recon(algorithm):
             psf = torch.rand(1, 32, 64, 3, dtype=torch_type)
             data = torch.rand(2, 1, 32, 64, 3, dtype=torch_type)
 
-            def pre_process(x, noise):
+            def pre_process(x, param):
                 return x
 
-            def post_process(x, noise):
+            def post_process(x, param, residual):
                 return x
 
             recon = algorithm(