From 21cc3de745364db836578397e3755a14cd0b283b Mon Sep 17 00:00:00 2001 From: Eric Bezzam Date: Tue, 9 Jul 2024 11:36:37 +0000 Subject: [PATCH] Fix test. --- test/test_algos.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/test/test_algos.py b/test/test_algos.py index e55b3147..441277fd 100644 --- 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(