mypy plot_warping_function

Author: robbievanleeuwen

src/sectionproperties/analysis/fea.py

@@ -280,7 +280,9 @@ def geometric_properties(
 
     def torsion_properties(
         self,
-    ) -> tuple[npt.NDArray[np.float64], npt.NDArray[np.float64], npt.NDArray[np.float64]]:
+    ) -> tuple[
+        npt.NDArray[np.float64], npt.NDArray[np.float64], npt.NDArray[np.float64]
+    ]:
         """Calculates the element warping stiffness matrix and the torsion load vector.
 
         Returns:

src/sectionproperties/analysis/section.py

@@ -1690,8 +1690,8 @@ def plot_warping_function(
         cmap: str = "viridis",
         alpha: float = 0.2,
         with_lines: bool = True,
-        **kwargs,
-    ):
+        **kwargs: Any,
+    ) -> matplotlib.axes.Axes:
         r"""Plots the warping function over the mesh.
 
         Args: