diff --git a/brainles_preprocessing/registration/__init__.py b/brainles_preprocessing/registration/__init__.py index b4c8bcb..4c941cc 100644 --- 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