Import DTI at the module level __init__.py

Author: arokem

nipype/interfaces/dipy/__init__.py

@@ -1,3 +1,3 @@
 from .tracks import TrackDensityMap
-from .tensors import TensorMode
+from .tensors import TensorMode, DTI
 from .preprocess import Resample, Denoise

nipype/interfaces/dipy/tensors.py

@@ -164,7 +164,7 @@ class TensorMode(BaseInterface):
     >>> mode.inputs.in_file = 'diffusion.nii'
     >>> mode.inputs.bvecs = 'bvecs'
     >>> mode.inputs.bvals = 'bvals'
-    >>> mode.inputs.mask = 'wm_mask.nii'
+    >>> mode.inputs.mask_file = 'wm_mask.nii'
     >>> mode.run()                                   # doctest: +SKIP
     """
     input_spec = TensorModeInputSpec

nipype/interfaces/dipy/tests/test_auto_DTI.py

@@ -9,7 +9,7 @@ def test_DTI_inputs():
     ),
     in_file=dict(mandatory=True,
     ),
-    mask_file=dict(mandatory=False, 
+    mask_file=dict(mandatory=False,
     ),
     out_filename=dict(genfile=True,
     ),

nipype/interfaces/dipy/tests/test_auto_TensorMode.py

@@ -9,6 +9,8 @@ def test_TensorMode_inputs():
     ),
     in_file=dict(mandatory=True,
     ),
+    mask_file=dict(mandatory=False,
+    ),
     out_filename=dict(genfile=True,
     ),
     )