Merge pull request #1161 from 36000/clean_bundles_space_fix

[FIX] Do not assume VOX space in clean bundles function
yeatmanlab · Sep 3, 2024 · fe30b28 · fe30b28
2 parents ef450a7 + ac547db
commit fe30b28
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/AFQ/recognition/cleaning.py b/AFQ/recognition/cleaning.py
@@ -175,7 +175,7 @@ def clean_bundle(tg, n_points=100, clean_rounds=5, distance_threshold=3,
 
     # Select based on the variable that was keeping track of things for us:
     if hasattr(tg, "streamlines"):
-        out = StatefulTractogram(tg.streamlines[idx], tg, Space.VOX)
+        out = StatefulTractogram(tg.streamlines[idx], tg, tg.space)
     else:
         out = streamlines[idx]
     if return_idx: