Apply suggestions from code review

Co-Authored-By: MichielCottaar <[email protected]>

Author: effigies

nibabel/cifti2/cifti2.py

@@ -1288,7 +1288,7 @@ def from_axes(cls, axes):
 
         Parameters
         ----------
-        axes : Tuple[cifti2_axes.Axis]
+        axes : tuple of :class`.cifti2_axes.Axis`
             sequence of Cifti2 axes describing each row/column of the matrix to be stored
 
         Returns
@@ -1328,7 +1328,7 @@ def __init__(self,
             Object containing image data.  It should be some object that
             returns an array from ``np.asanyarray``.  It should have a
             ``shape`` attribute or property.
-        header : Cifti2Header instance or Sequence[cifti2_axes.Axis]
+        header : Cifti2Header instance or sequence of :class:`cifti2_axes.Axis`
             Header with data for / from XML part of CIFTI-2 format.
             Alternatively a sequence of cifti2_axes.Axis objects can be provided
             describing each dimension of the array.

nibabel/cifti2/cifti2_axes.py

@@ -124,11 +124,11 @@ def from_index_mapping(mim):
 
     Parameters
     ----------
-    mim : cifti2.Cifti2MatrixIndicesMap
+    mim : :class:`.cifti2.Cifti2MatrixIndicesMap`
 
     Returns
     -------
-    subtype of Axis
+    subclass of :class:`Axis`
     """
     return_type = {'CIFTI_INDEX_TYPE_SCALARS': ScalarAxis,
                    'CIFTI_INDEX_TYPE_LABELS': LabelAxis,
@@ -149,7 +149,7 @@ def to_header(axes):
 
     Returns
     -------
-    cifti2.Cifti2Header
+    :class:`.cifti2.Cifti2Header`
     """
     axes = tuple(axes)
     mims_all = []
@@ -777,7 +777,7 @@ def from_brain_models(cls, named_brain_models):
         for idx_parcel, (parcel_name, bm) in enumerate(named_brain_models):
             all_names.append(parcel_name)
 
-            voxels = bm.voxel[~bm.surface_mask]
+            voxels = bm.voxel[bm.volume_mask]
             if voxels.shape[0] != 0:
                 if affine is None:
                     affine = bm.affine