GradICON requires the itk_wrapper

uncbiag · Dec 20, 2023 · 452764f · 452764f
1 parent 50d2cba
commit 452764f
Showing 1 changed file with 6 additions and 13 deletions.
diff --git a/src/icon_registration/__init__.py b/src/icon_registration/__init__.py
@@ -1,16 +1,7 @@
-from icon_registration.losses import (
-    LNCC,
-    LNCCOnlyInterpolated,
-    BlurredSSD,
-    GradientICON,
-    InverseConsistentNet,
-    gaussian_blur,
-    ssd_only_interpolated,
-    ssd,
-    SSDOnlyInterpolated,
-    SSD,
-    NCC
-)
+from icon_registration.losses import (LNCC, LNCCOnlyInterpolated, BlurredSSD,
+                                      GradientICON, InverseConsistentNet,
+                                      gaussian_blur, ssd_only_interpolated,
+                                      ssd, SSDOnlyInterpolated, SSD, NCC)
 from icon_registration.network_wrappers import (
     DownsampleRegistration,
     FunctionFromMatrix,
@@ -19,3 +10,5 @@
     TwoStepRegistration,
 )
 from icon_registration.train import train_batchfunction, train_datasets
+
+import icon_registration.itk_wrapper