Address comments

apaszke · apaszke · commit 29ffef671b54 · 2020-03-06T11:38:24.000+01:00
diff --git a/lib/matplotlib/cbook/__init__.py b/lib/matplotlib/cbook/__init__.py
@@ -2003,9 +2003,9 @@ def _array_perimeter(arr):
 
 
 def _unfold(arr, axis, size, step):
-    """Append an extra dimension containing sliding windows along ``axis``.
+    """Append an extra dimension containing sliding windows along **axis**.
 
-    All windows are of size ``size`` and begin with every ``step`` elements.
+    All windows are of size **size** and begin with every **step** elements.
 
     Parameters
     ----------
@@ -2027,10 +2027,19 @@ def _unfold(arr, axis, size, step):
 
 
 def _array_patch_perimeters(x, rstride, cstride):
-    """Extract perimeters of patches from ``arr``.
+    """Extract perimeters of patches from **arr**.
 
-    Extracted patches are of size ``(rstride + 1) x (cstride + 1)`` and share
+    Extracted patches are of size (**rstride** + 1) x (**cstride** + 1) and share
     perimeters with their neighbors.
+
+    Parameters
+    ----------
+    arr : ndarray, shape (N, M)
+
+    Returns
+    -------
+    patches : ndarray, shape (N / rstride * M / cstride,
+                              2 * (**rstride** + **cstride**))
     """
     assert rstride > 0 and cstride > 0
     assert (x.shape[0] - 1) % rstride == 0
diff --git a/lib/mpl_toolkits/mplot3d/axes3d.py b/lib/mpl_toolkits/mplot3d/axes3d.py
@@ -1562,10 +1562,10 @@ def plot_surface(self, X, Y, Z, *args, norm=None, vmin=None,
         if (rows - 1) % rstride == 0 and \
            (cols - 1) % cstride == 0 and \
            fcolors is None:
-            polys = np.stack([cbook._array_patch_perimeters(a, rstride,
-                                                            cstride)
-                              for a in (X, Y, Z)],
-                             axis=-1)
+            polys = np.stack(
+                [cbook._array_patch_perimeters(a, rstride, cstride)
+                 for a in (X, Y, Z)],
+                axis=-1)
         else:
             # evenly spaced, and including both endpoints
             row_inds = list(range(0, rows-1, rstride)) + [rows-1]
