Move import of recon methods to module scope

DiamondLightSource · Aug 14, 2024 · 649a558 · 649a558
1 parent 624f482
commit 649a558
Showing 1 changed file with 3 additions and 4 deletions.
diff --git 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: