handle_cache for fit_transform

Author: clementpoiret

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "ROILoc"
-version = "0.2.7"
+version = "0.2.8"
 description = "A simple package to center and crop T1w & T2w MRIs around a given region of interest by its name."
 license = "MIT"
 readme = "README.rst"

roiloc/__init__.py

@@ -1 +1 @@
-__version__ = "0.2.7"
+__version__ = "0.2.8"

roiloc/locator.py

@@ -121,16 +121,18 @@ def transform(self, image: ANTsImage) -> list:
             for side in ["right", "left"]
         ]
 
-    def fit_transform(self, image: ANTsImage) -> list:
+    @handle_cache
+    def fit_transform(self, image: ANTsImage, outprefix: str = "") -> list:
         """Fit the ROI to the image and transform.
 
         Args:
             image (ANTsImage): Image to fit the ROI to.
+            outprefix (str, optional): Prefix for ANTs' temporary files.
 
         Returns:
             list: List of transformed images.
         """
-        self.fit(image)
+        self.fit(image, outprefix=outprefix)
         return self.transform(image)
 
     def inverse_transform(self,

tests/test_roiloc.py

@@ -2,4 +2,4 @@
 
 
 def test_version():
-    assert __version__ == '0.2.7'
+    assert __version__ == '0.2.8'