import

Signed-off-by: neuronflow <[email protected]>
BrainLesion · Mar 30, 2024 · f18917b · f18917b
1 parent 11e3012
commit f18917b
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/brainles_preprocessing/registration/__init__.py b/brainles_preprocessing/registration/__init__.py
@@ -7,4 +7,12 @@
         "ANTS package not found. If you want to use it, please install it using 'pip install brainles_preprocessing[ants]'"
     )
 
+try:
+    from .eReg.eReg import eRegRegistrator
+except ImportError:
+    warnings.warn(
+        "eReg package not found. If you want to use it, please install it using 'pip install brainles_preprocessing[ereg]'"
+    )
+
+
 from .niftyreg.niftyreg import NiftyRegRegistrator