From 53ec5db3dd1ecd92668a69b85a1a57e364f93bc8 Mon Sep 17 00:00:00 2001 From: John K Date: Tue, 30 Jun 2020 09:27:47 -0700 Subject: [PATCH] make_bundle_dict should only be called after reg_template is settled --- AFQ/api.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/AFQ/api.py b/AFQ/api.py index 417735c43..7c4a515fb 100644 --- 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}*"))