Update dmriprep/utils/vectors.py

Co-Authored-By: Ariel Rokem <arokem@gmail.com>
nipreps · Mar 24, 2020 · 49e0a0a · 49e0a0a
1 parent 66f8b48
commit 49e0a0a
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/dmriprep/utils/vectors.py b/dmriprep/utils/vectors.py
@@ -456,6 +456,10 @@ def _rasb_to_bvec_list(in_rasb):
     Returns
     -------
     List of b-vectors as floats.
+    
+    Returns
+    --------
+    list : each element is a b-vector
     """
     import numpy as np
 
@@ -480,4 +484,4 @@ def _rasb_to_bval_floats(in_rasb):
     import numpy as np
 
     ras_b_mat = np.genfromtxt(in_rasb, delimiter="\t")
-    return [float(bval) for bval in ras_b_mat[:, 3] if bval > 0]
+    return [float(bval) for bval in ras_b_mat[:, 3] if bval > 0]