lazily import medimages4tests to speed up cases where it isn't required

Australian-Imaging-Service · Oct 10, 2024 · 7d8f430 · 7d8f430
1 parent c0eedb3
commit 7d8f430
Showing 1 changed file with 13 additions and 10 deletions.
diff --git a/xnat4tests/data.py b/xnat4tests/data.py
@@ -6,16 +6,6 @@
 from contextlib import contextmanager
 from pathlib import Path
 from xnat.exceptions import XNATResponseError
-from medimages4tests.dummy.dicom.mri.t1w.siemens.skyra.syngo_d13c import (
-    get_image as t1w_syngo,
-)
-from medimages4tests.dummy.dicom.mri.dwi.siemens.skyra.syngo_d13c import (
-    get_image as dwi_syngo,
-)
-from medimages4tests.dummy.dicom.mri.fmap.siemens.skyra.syngo_d13c import (
-    get_image as fmap_syngo,
-)
-from medimages4tests.mri.neuro.t1w import get_image as openneuro_t1w
 from .base import connect
 from .config import Config
 from .utils import logger
@@ -64,6 +54,15 @@ def add_data(dataset: str, config_name: str or dict = "default"):
         )
 
     if dataset == "dummydicom":
+        from medimages4tests.dummy.dicom.mri.t1w.siemens.skyra.syngo_d13c import (
+            get_image as t1w_syngo,
+        )
+        from medimages4tests.dummy.dicom.mri.dwi.siemens.skyra.syngo_d13c import (
+            get_image as dwi_syngo,
+        )
+        from medimages4tests.dummy.dicom.mri.fmap.siemens.skyra.syngo_d13c import (
+            get_image as fmap_syngo,
+        )
 
         _upload_dicom_data(
             [t1w_syngo(), dwi_syngo(), fmap_syngo()],
@@ -74,6 +73,7 @@ def add_data(dataset: str, config_name: str or dict = "default"):
         )
 
     elif dataset == "openneuro-t1w":
+        from medimages4tests.mri.neuro.t1w import get_image as openneuro_t1w
 
         _upload_directly(
             {"t1w": openneuro_t1w()},
@@ -119,6 +119,9 @@ def add_data(dataset: str, config_name: str or dict = "default"):
         )
 
     elif dataset == "user-training":
+        from medimages4tests.dummy.dicom.mri.fmap.siemens.skyra.syngo_d13c import (
+            get_image as fmap_syngo,
+        )
 
         _upload_dicom_data(
             [t1w_syngo(), fmap_syngo()],