Use same instead of img with load_tractogram to avoid bug where dipy …

…only looks at srow to get affine
yeatmanlab · Aug 25, 2020 · d942d41 · d942d41
1 parent 595bee7
commit d942d41
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/AFQ/api.py b/AFQ/api.py
@@ -858,7 +858,7 @@ def _segment(self, row):
             streamlines_file = self._streamlines(row)
 
             img = nib.load(row['dwi_file'])
-            tg = load_tractogram(streamlines_file, img, Space.VOX)
+            tg = load_tractogram(streamlines_file, "same", Space.VOX)
             if self.use_prealign:
                 reg_prealign = np.load(self._reg_prealign(row))
             else:
@@ -902,7 +902,7 @@ def _clean_bundles(self, row):
             bundles_file = self._segment(row)
 
             sft = load_tractogram(bundles_file,
-                                  row['dwi_img'],
+                                  "same",
                                   Space.VOX)
 
             tgram = nib.streamlines.Tractogram([], {'bundle': []})