fix: safe import for gpunufft.

paquiteau · Dec 18, 2023 · 0a7be76 · 0a7be76
1 parent 905001b
commit 0a7be76
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/simfmri/handlers/acquisition/workers.py b/src/simfmri/handlers/acquisition/workers.py
@@ -9,7 +9,12 @@
 from fmri.operators.fourier import FFT_Sense
 from joblib import Parallel, delayed
 from mrinufft import get_operator
-from mrinufft.operators.interfaces.gpunufft import make_pinned_smaps
+
+try:
+    from mrinufft.operators.interfaces.gpunufft import make_pinned_smaps
+except ImportError:
+    make_pinned_smaps = None
+
 from tqdm.auto import tqdm
 
 from simfmri.simulation import SimData