fix: correct define FFTFilters upon import

RI-imaging · Nov 8, 2024 · 2361769 · 2361769
1 parent b812c07
commit 2361769
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/qpretrieve/fourier/__init__.py b/qpretrieve/fourier/__init__.py
@@ -11,10 +11,14 @@
 
 try:
     from .ff_cupy import FFTFilterCupy
-    from .ff_cupy3D import FFTFilterCupy3D
 except ImportError:
     FFTFilterCupy = None
 
+try:
+    from .ff_cupy3D import FFTFilterCupy3D
+except ImportError:
+    FFTFilterCupy3D = None
+
 PREFERRED_INTERFACE = None