env

sgoldenlab · Nov 22, 2024 · 4c1daf9 · 4c1daf9
1 parent c8a2c70
commit 4c1daf9
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/simba/data_processors/cuda/image.py b/simba/data_processors/cuda/image.py
@@ -19,7 +19,10 @@
 
 import cv2
 import numpy as np
-from cupyx.scipy.ndimage import rotate
+try:
+    from cupyx.scipy.ndimage import rotate
+except:
+    from scipy.ndimage import rotate
 from numba import cuda
 
 from simba.data_processors.cuda.utils import _cuda_mse