FIX: Ensure templates are pulled on workflow construction

Author: mgxd

smriprep/utils/misc.py

@@ -21,6 +21,11 @@
 #     https://www.nipreps.org/community/licensing/
 #
 """Self-contained utilities to be used within Function nodes."""
+from pathlib import Path
+import typing as ty
+
+if ty.TYPE_CHECKING:
+    from niworkflows.utils.spaces import Reference
 
 
 def apply_lut(in_dseg, lut, newpath=None):
@@ -95,3 +100,25 @@ def fs_isRunning(subjects_dir, subject_id, mtime_tol=86400, logger=None):
     if logger:
         logger.warn(f'Removed "IsRunning*" files found under {subj_dir}')
     return subjects_dir
+
+
+def get_template_t1w(template: str, sloppy: bool = False) -> Path:
+    """Query templateflow for the T1w to ensure it is present on the filesystem."""
+    import templateflow.api as tf
+
+    spec = {}
+    _space = template.split(':', 1)
+    if len(_space) > 1:
+        spec['cohort'] = _space[1].replace('cohort-', '')
+    space = _space[0]
+
+    available_res = tf.TF_LAYOUT.get_resolutions(template=space)
+    if sloppy and 2 in available_res:
+        res = 2
+    elif 1 in available_res:
+        res = 1
+    else:
+        res = None
+    spec['resolution'] = res
+
+    return tf.get(space, desc=None, suffix='T1w', **spec)

smriprep/workflows/anatomical.py

@@ -973,6 +973,9 @@ def init_anat_fit_wf(
     templates = []
     found_xfms = {}
     for template in spaces.get_spaces(nonstandard=False, dim=(3,)):
+        from ..utils.misc import get_template_t1w
+
+        get_template_t1w(template, sloppy)
         xfms = precomputed.get('transforms', {}).get(template, {})
         if set(xfms) != {'forward', 'reverse'}:
             templates.append(template)