Fix test.

LCAV · Jul 9, 2024 · 21cc3de · 21cc3de
1 parent 4d97665
commit 21cc3de
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/test/test_algos.py b/test/test_algos.py
@@ -172,7 +172,7 @@ def test_trainable_recon(algorithm):
             def pre_process(x, param):
                 return x
 
-            def post_process(x, param, residual):
+            def post_process(x, param, residual=None):
                 return x
 
             recon = algorithm(
@@ -206,10 +206,10 @@ def test_trainable_batch(algorithm):
         data2 = torch.rand(1, 1, 34, 64, 3, dtype=torch_type)
         data2[0, 0, ...] = data1[0, 0, ...]
 
-        def pre_process(x, noise):
+        def pre_process(x, param):
             return x
 
-        def post_process(x, noise):
+        def post_process(x, param, residual=None):
             return x
 
         recon = algorithm(