add tip_iteration to autotrack

PennLINC · Jan 12, 2025 · 8509d7f · 8509d7f
1 parent 8b76ea3
commit 8509d7f
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/qsirecon/interfaces/dsi_studio.py b/qsirecon/interfaces/dsi_studio.py
@@ -713,7 +713,19 @@ class _AutoTrackInputSpec(DSIStudioCommandLineInputSpec):
     output_dir = traits.Str(
         "cwd", argstr="%s", usedefault=True, desc="Forces DSI Studio to write results in cwd"
     )
-    _boilerplate_traits = ["track_id", "track_voxel_ratio", "tolerance", "yield_rate"]
+    tip_iterations = traits.Int(
+        16,
+        usedefault=False,
+        desc="Topologically-informed pruning iterations",
+        argstr="--tip_iteration=%d",
+    )
+    _boilerplate_traits = [
+        "track_id",
+        "track_voxel_ratio",
+        "tolerance",
+        "yield_rate",
+        "tip_iteration",
+    ]
 
 
 class _AutoTrackOutputSpec(TraitedSpec):