Skip to content

Commit 0bfabab

Browse files
Apply suggestions from code review
Co-authored-by: Chris Markiewicz <effigies@gmail.com>
1 parent 450593d commit 0bfabab

File tree

1 file changed

+9
-5
lines changed

1 file changed

+9
-5
lines changed

nipype/interfaces/mrtrix/convert.py

Lines changed: 9 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,10 @@ def get_data_dims(volume):
4040

4141

4242
def transform_to_affine(streams, header, affine):
43-
from dipy.tracking.utils import transform_tracking_output
43+
try:
44+
from dipy.tracking.utils import transform_tracking_output
45+
except ImportError:
46+
from dipy.tracking.utils import move_streamlines as transform_tracking_output
4447

4548
rotation, scale = np.linalg.qr(affine)
4649
streams = transform_tracking_output(streams, rotation)
@@ -193,10 +196,11 @@ class MRTrix2TrackVis(DipyBaseInterface):
193196
output_spec = MRTrix2TrackVisOutputSpec
194197

195198
def _run_interface(self, runtime):
196-
from dipy.tracking.utils import (
197-
affine_from_fsl_mat_file,
198-
transform_tracking_output,
199-
)
199+
from dipy.tracking.utils import affine_from_fsl_mat_file
200+
try:
201+
from dipy.tracking.utils import transform_tracking_output
202+
except ImportError:
203+
from dipy.tracking.utils import move_streamlines as transform_tracking_output
200204

201205
dx, dy, dz = get_data_dims(self.inputs.image_file)
202206
vx, vy, vz = get_vox_dims(self.inputs.image_file)

0 commit comments

Comments
 (0)