Merge pull request yeatmanlab#281 from 36000/template_resample_bug

make_bundle_dict should only be called after reg_template is settled
richford · Jul 1, 2020 · 0531df2 · 0531df2
2 parents 60dedef + 53ec5db
commit 0531df2
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/AFQ/api.py b/AFQ/api.py
@@ -298,9 +298,6 @@ def __init__(self,
 
         self.segmentation_params = default_seg_params
         self.seg_algo = self.segmentation_params["seg_algo"].lower()
-        self.bundle_dict = make_bundle_dict(bundle_names=bundle_names,
-                                            seg_algo=self.seg_algo,
-                                            resample_to=reg_template)
 
         default_clean_params = get_default_args(seg.clean_bundle)
         if clean_params is not None:
@@ -315,6 +312,11 @@ def __init__(self,
             if not isinstance(reg_template, nib.Nifti1Image):
                 reg_template = nib.load(reg_template)
             self.reg_template = reg_template
+
+        self.bundle_dict = make_bundle_dict(bundle_names=bundle_names,
+                                            seg_algo=self.seg_algo,
+                                            resample_to=reg_template)
+
         # This is the place in which each subject's full data lives
         self.dmriprep_dirs = glob.glob(op.join(dmriprep_path,
                                                f"{sub_prefix}*"))