Merge pull request #2 from NOEL-MNI/tf_conditional

add conditional import for tensorflow
NOEL-MNI · Sep 14, 2023 · ccabb5f · ccabb5f
2 parents b6006e7 + 217907a
commit ccabb5f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/app/utils/image_processing.py b/app/utils/image_processing.py
@@ -8,8 +8,8 @@
 import multiprocessing
 import numpy as np
 # import zipfile
-
-from antspynet.utilities import brain_extraction
+if os.environ.get("BRAIN_MASKING") == "cpu":
+    from antspynet.utilities import brain_extraction
 from .deepmask import *
 from .helpers import *
 from matplotlib.backends.backend_pdf import PdfPages