From 649a5581db7fa0067e5c40ced629f0cfb1923e2f Mon Sep 17 00:00:00 2001 From: Timothy Poon Date: Wed, 14 Aug 2024 16:46:30 +0100 Subject: [PATCH] Move import of recon methods to module scope --- httomolibgpu/recon/algorithm.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/httomolibgpu/recon/algorithm.py b/httomolibgpu/recon/algorithm.py index e0eebee9..09587258 100644 --- a/httomolibgpu/recon/algorithm.py +++ b/httomolibgpu/recon/algorithm.py @@ -29,6 +29,9 @@ from numpy import float32, complex64 from typing import Optional, Type +from tomobar.methodsDIR_CuPy import RecToolsDIRCuPy +from tomobar.methodsIR_CuPy import RecToolsIRCuPy + __all__ = [ "FBP", "LPRec", @@ -360,8 +363,6 @@ def _instantiate_direct_recon_class( Returns: Type[RecToolsDIRCuPy]: an instance of the direct recon class """ - from tomobar.methodsDIR_CuPy import RecToolsDIRCuPy - if center is None: center = data.shape[2] // 2 # making a crude guess if recon_size is None: @@ -400,8 +401,6 @@ def _instantiate_iterative_recon_class( Returns: Type[RecToolsIRCuPy]: an instance of the iterative class """ - from tomobar.methodsIR_CuPy import RecToolsIRCuPy - if center is None: center = data.shape[2] // 2 # making a crude guess if recon_size is None: