Make ants an optional dependency

BrainLesion · Mar 1, 2024 · d2e2227 · d2e2227
1 parent 8060dcc
commit d2e2227
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 2 deletions.
diff --git a/brainles_preprocessing/registration/__init__.py b/brainles_preprocessing/registration/__init__.py
@@ -1,2 +1,10 @@
-from .ANTs.ANTs import ANTsRegistrator
+import warnings
+
+try:
+    from .ANTs.ANTs import ANTsRegistrator
+except ImportError:
+    warnings.warn(
+        "ANTS package not found. If you want to use it, please install it using 'pip install antspyx'"
+    )
+
 from .niftyreg.niftyreg import NiftyRegRegistrator
diff --git a/pyproject.toml b/pyproject.toml
@@ -40,7 +40,6 @@ classifiers = [
   "Operating System :: OS Independent",
 ]
 
-
 [tool.poetry.dependencies]
 python = "^3.10"
 # core
@@ -58,6 +57,12 @@ tqdm = "^4.64.1"
 auxiliary = "^0.0.41"
 rich = "^13.6.0"
 
+
+antspyx = { version = "^0.4.2", optional = true }
+
+[tool.poetry.extras]
+ants = ["antspyx"]
+
 [tool.poetry.dev-dependencies]
 pytest = "^6.2"