moving all gpu related imports under try

Author: dkazanc

httomolibgpu/cupywrapper.py

@@ -2,7 +2,6 @@
 try:
     import cupy as cp
     import nvtx
-    import cupyx
 
     try:
         cp.cuda.Device(0).compute_capability
@@ -18,4 +17,3 @@
     import numpy as cp
 
     nvtx = Mock()
-    cupyx = Mock()

httomolibgpu/misc/corr.py

@@ -31,7 +31,13 @@
 nvtx = cupywrapper.nvtx
 
 from numpy import float32
-from httomolibgpu.cuda_kernels import load_cuda_module
+from unittest.mock import Mock
+
+try:
+    from httomolibgpu.cuda_kernels import load_cuda_module
+except ImportError as e:
+    load_cuda_module = Mock()
+
 
 __all__ = [
     "median_filter",

httomolibgpu/misc/morph.py

@@ -25,9 +25,13 @@
 
 cp = cupywrapper.cp
 nvtx = cupywrapper.nvtx
-cupyx = cupywrapper.cupyx
 
-from cupyx.scipy.interpolate import interpn
+from unittest.mock import Mock
+
+try:
+    from cupyx.scipy.interpolate import interpn
+except ImportError as e:
+    interpn = Mock()
 
 from typing import Literal
 

httomolibgpu/prep/alignment.py

@@ -25,9 +25,13 @@
 
 cp = cupywrapper.cp
 nvtx = cupywrapper.nvtx
-cupyx = cupywrapper.cupyx
 
-from cupyx.scipy.ndimage import map_coordinates
+from unittest.mock import Mock
+
+try:
+    from cupyx.scipy.ndimage import map_coordinates
+except ImportError as e:
+    map_coordinates = Mock()
 
 from typing import Dict, List
 

httomolibgpu/prep/normalize.py

@@ -26,7 +26,12 @@
 cp = cupywrapper.cp
 nvtx = cupywrapper.nvtx
 
-from cupy import mean
+from unittest.mock import Mock
+
+try:
+    from cupy import mean
+except ImportError as e:
+    mean = Mock()
 
 from numpy import float32
 from typing import Tuple

httomolibgpu/prep/phase.py

@@ -26,8 +26,15 @@
 cp = cupywrapper.cp
 nvtx = cupywrapper.nvtx
 
-from httomolibgpu.cuda_kernels import load_cuda_module
-from cupyx.scipy.fft import fft2, ifft2, fftshift
+from unittest.mock import Mock
+try:
+    from httomolibgpu.cuda_kernels import load_cuda_module
+    from cupyx.scipy.fft import fft2, ifft2, fftshift
+except ImportError as e:
+    load_cuda_module = Mock()
+    fft2 = Mock()
+    ifft2 = Mock()
+    fftshift = Mock()
 
 from numpy import float32
 from typing import Union

httomolibgpu/prep/stripe.py

@@ -26,7 +26,13 @@
 cp = cupywrapper.cp
 nvtx = cupywrapper.nvtx
 
-from cupyx.scipy.ndimage import median_filter, binary_dilation, uniform_filter1d
+from unittest.mock import Mock
+try:
+    from cupyx.scipy.ndimage import median_filter, binary_dilation, uniform_filter1d    
+except ImportError as e:
+    median_filter = Mock()
+    binary_dilation = Mock()
+    uniform_filter1d = Mock()
 
 from typing import Union
 

httomolibgpu/recon/algorithm.py

@@ -26,8 +26,13 @@
 cp = cupywrapper.cp
 nvtx = cupywrapper.nvtx
 
-from tomobar.methodsDIR_CuPy import RecToolsDIRCuPy
-from tomobar.methodsIR_CuPy import RecToolsIRCuPy
+from unittest.mock import Mock
+try:
+    from tomobar.methodsDIR_CuPy import RecToolsDIRCuPy
+    from tomobar.methodsIR_CuPy import RecToolsIRCuPy
+except ImportError as e:
+    RecToolsDIRCuPy = Mock()
+    RecToolsIRCuPy = Mock()
 
 from numpy import float32, complex64
 from typing import Optional, Type

httomolibgpu/recon/rotation.py

@@ -26,11 +26,21 @@
 cp = cupywrapper.cp
 nvtx = cupywrapper.nvtx
 
-from httomolibgpu.cuda_kernels import load_cuda_module
-from cupyx.scipy.ndimage import shift, gaussian_filter
-from skimage.registration import phase_cross_correlation
-from cupyx.scipy.fftpack import get_fft_plan
-from cupyx.scipy.fft import rfft2
+
+from unittest.mock import Mock
+try:
+    from httomolibgpu.cuda_kernels import load_cuda_module
+    from cupyx.scipy.ndimage import shift, gaussian_filter
+    from skimage.registration import phase_cross_correlation
+    from cupyx.scipy.fftpack import get_fft_plan
+    from cupyx.scipy.fft import rfft2    
+except ImportError as e:
+    load_cuda_module = Mock()
+    shift = Mock()
+    gaussian_filter = Mock()
+    phase_cross_correlation = Mock()
+    get_fft_plan = Mock()
+    rfft2 = Mock()
 
 import math
 from typing import List, Literal, Optional, Tuple, Union