Fix mistyped function method and fix stub

trananso · trananso · commit b12b7b0e4b8e · 2024-05-06T16:39:10.000-04:00
diff --git a/lib/matplotlib/transforms.py b/lib/matplotlib/transforms.py
@@ -81,40 +81,6 @@ def strrepr(x): return repr(x) if isinstance(x, str) else str(x)
         + ")")
 
 
-def matrix_transform(vertices, mtx):
-    """
-    Transforms a vertex or set of vertices with a matrix mtx of
-    one dimension higher.
-
-    Parameters
-    ----------
-    vertices : n-element array or (m, n) array, with m vertices
-    mtx      : (n+1, n+1) matrix
-    """
-    values = np.asanyarray(vertices)
-    _, input_dims = mtx.shape
-    input_dims = input_dims - 1
-
-    if (len(values.shape) == 1):
-        # single point
-        if (values.shape == (input_dims,)):
-            point = mtx.dot(np.append(values, [1]))
-            point = point/point[-1]
-            return point[:input_dims]
-        raise RuntimeError("Invalid vertices provided to transform")
-
-    # multiple points
-    if (len(values.shape) == 2 and values.shape[1] == input_dims):
-        points = np.hstack((values, np.ones((values.shape[0], 1))))
-        points = np.dot(mtx, points.T).T
-        last_coords = points[:, -1]
-        points = points / last_coords[:, np.newaxis]
-        return points[:, :-1]
-
-    raise ValueError("Dimensions of input must match the input dimensions of "
-                     "the transform")
-
-
 class TransformNode:
     """
     The base class for anything that participates in the transform tree
@@ -2361,7 +2327,7 @@ def rotate_around_vector(self, vector, theta):
         np.matmul(rot, self._mtx, out=self._mtx)
         return self
 
-    def rotate_around_vector_deg(self, vector, degrees):
+    def rotate_deg_around_vector(self, vector, degrees):
         """
         Add a rotation (in radians) around the vector (vx, vy, vz) in place.
 
@@ -3377,3 +3343,37 @@ def offset_copy(trans, fig=None, x=0.0, y=0.0, units='inches'):
         y /= 72.0
     # Default units are 'inches'
     return trans + ScaledTranslation(x, y, fig.dpi_scale_trans)
+
+
+def matrix_transform(vertices, mtx):
+    """
+    Transforms a vertex or set of vertices with a matrix mtx of
+    one dimension higher.
+
+    Parameters
+    ----------
+    vertices : n-element array or (m, n) array, with m vertices
+    mtx      : (n+1, n+1) matrix
+    """
+    values = np.asanyarray(vertices)
+    _, input_dims = mtx.shape
+    input_dims = input_dims - 1
+
+    if (len(values.shape) == 1):
+        # single point
+        if (values.shape == (input_dims,)):
+            point = mtx.dot(np.append(values, [1]))
+            point = point/point[-1]
+            return point[:input_dims]
+        raise RuntimeError("Invalid vertices provided to transform")
+
+    # multiple points
+    if (len(values.shape) == 2 and values.shape[1] == input_dims):
+        points = np.hstack((values, np.ones((values.shape[0], 1))))
+        points = np.dot(mtx, points.T).T
+        last_coords = points[:, -1]
+        points = points / last_coords[:, np.newaxis]
+        return points[:, :-1]
+
+    raise ValueError("Dimensions of input must match the input dimensions of "
+                     "the transform")
diff --git a/lib/matplotlib/transforms.pyi b/lib/matplotlib/transforms.pyi
@@ -364,3 +364,7 @@ def offset_copy(
     y: float = ...,
     units: Literal["inches", "points", "dots"] = ...,
 ) -> Transform: ...
+def matrix_transform(
+    vertices: ArrayLike,
+    mtx: ArrayLike
+) -> ArrayLike: ...
