Pass kwargs to Brain_Data nifti masker on init (e.g. for caching)

cosanlab · May 30, 2022 · 6ad1db6 · 6ad1db6
1 parent 24d8cc4
commit 6ad1db6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nltools/data/brain_data.py b/nltools/data/brain_data.py
@@ -114,7 +114,7 @@ def __init__(self, data=None, Y=None, X=None, mask=None, **kwargs):
             # Load default mask
             self.mask = nib.load(resolve_mni_path(MNI_Template)["mask"])
         # Learn transformation on mask
-        self.nifti_masker = NiftiMasker(mask_img=self.mask)
+        self.nifti_masker = NiftiMasker(mask_img=self.mask, **kwargs)
 
         if data is not None:
             if isinstance(data, str) or isinstance(data, Path):