From 8ee8bf9237d685ab93ed7ed0a98037c013e592a0 Mon Sep 17 00:00:00 2001 From: Abe Levitan Date: Fri, 3 Apr 2026 18:59:09 +0200 Subject: [PATCH] Fix bug where fancyptycho test was not being run on GPU --- tests/models/test_fancy_ptycho.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/models/test_fancy_ptycho.py b/tests/models/test_fancy_ptycho.py index 93d089d..f04c751 100644 --- a/tests/models/test_fancy_ptycho.py +++ b/tests/models/test_fancy_ptycho.py @@ -97,8 +97,8 @@ def test_lab_ptycho(lab_ptycho_cxi, reconstruction_device, show_plot): print('Running reconstruction on provided reconstruction_device,', reconstruction_device) - #model.to(device=reconstruction_device) - #dataset.get_as(device=reconstruction_device) + model.to(device=reconstruction_device) + dataset.get_as(device=reconstruction_device) for loss in model.Adam_optimize(50, dataset, lr=0.02, batch_size=10): print(model.report())