enh: removed straneous comments, update nonlinear

nipy · Apr 26, 2024 · ab28efc · ab28efc
1 parent 748e1d3
commit ab28efc
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/nitransforms/nonlinear.py b/nitransforms/nonlinear.py
@@ -163,7 +163,7 @@ def map(self, x, inverse=False):
         indexes = np.round(ijk).astype("int")
 
         if np.all(np.abs(ijk - indexes) < 1e-3):
-            indexes = tuple(tuple(i) for i in indexes.T)
+            indexes = tuple(tuple(i) for i in indexes)
             return self._field[indexes]
 
         return np.vstack(tuple(

diff --git a/nitransforms/tests/test_base.py b/nitransforms/tests/test_base.py
@@ -42,9 +42,7 @@ def test_ImageGrid(get_testdata, image_orientation):
     # Test ras2vox and vox2ras conversions
     ijk = [[10, 10, 10], [40, 4, 20], [0, 0, 0], [s - 1 for s in im.shape[:3]]]
     xyz = [img._affine.dot(idx + [1])[:-1] for idx in ijk]
-    # xyz = np.array([np.tensordot(img._affine, idx + [1], axes=1)[:-1] for idx in ijk])
 
-    # import pdb; pdb.set_trace()
     assert np.allclose(np.squeeze(img.ras(ijk[0])), xyz[0])
     assert np.allclose(np.round(img.index(xyz[0])), ijk[0])
     assert np.allclose(img.ras(ijk).T, xyz)