fix: new mypy errors due to numpy typing

Author: alexfikl

modepy/modal_decay.py

@@ -103,18 +103,18 @@ def make_mode_number_vector(
     return mode_number_vector
 
 
-def create_decay_baseline(mode_number_vector: np.ndarray, n: int) -> np.ndarray:
+def create_decay_baseline(
+        mode_number_vector: np.ndarray[tuple[int, ...], np.dtype[np.floating]],
+        n: int) -> np.ndarray[tuple[int, ...], np.dtype[np.floating]]:
     """Create a vector of modal coefficients that exhibit 'optimal'
     (:math:`k^{-N}`) decay.
     """
 
     zeros = mode_number_vector == 0
 
     modal_coefficients = mode_number_vector**(-n)
-    modal_coefficients[zeros] = 1  # irrelevant, just keeps log from NaNing
-
-    # NOTE: mypy seems to be confused by the __itruediv__ argument types
-    modal_coefficients /= la.norm(modal_coefficients)   # type: ignore[arg-type,misc]
+    modal_coefficients[zeros] = 1.0  # irrelevant, just keeps log from NaNing
+    modal_coefficients /= la.norm(modal_coefficients)
 
     return modal_coefficients
 

modepy/quadrature/clenshaw_curtis.py

@@ -49,7 +49,7 @@ def _make_clenshaw_curtis_nodes_and_weights(n: int) -> tuple[np.ndarray, np.ndar
     # Clenshaw-Curtis weights
     w = np.concatenate([2 / N / (N - 2), 1 / N[-1:], np.zeros(m)])
     w = 0 - w[:-1] - w[-1:0:-1]
-    g0 = -np.ones(n)
+    g0: np.ndarray[tuple[int, ...], np.dtype[np.floating]] = -np.ones(n)
     g0[r] = g0[r] + n
     g0[m] = g0[m] + n
     g0 = g0 / (n**2 - 1 + (n % 2))

modepy/quadrature/jacobi_gauss.py

@@ -50,7 +50,6 @@
 THE SOFTWARE.
 """
 
-
 import numpy as np
 import numpy.linalg as la
 
@@ -267,7 +266,7 @@ def jacobi_gauss_lobatto_nodes(
     Exact to degree :math:`2N - 3`.
     """
 
-    x = np.zeros((N + 1,))
+    x: np.ndarray[tuple[int, ...], np.dtype[np.floating]] = np.zeros((N + 1,))
     if N == 0:
         x[0] = 0
         return x