Appease mypy

Author: jessegrabowski

pytensor/link/numba/dispatch/linalg/decomposition/lu_factor.py

@@ -60,9 +60,7 @@ def impl(
     return impl
 
 
-def _lu_factor(
-    A: np.ndarray, overwrite_a: bool = False
-) -> tuple[np.ndarray, np.ndarray]:
+def _lu_factor(A: np.ndarray, overwrite_a: bool = False):
     """
     Thin wrapper around scipy.linalg.lu_factor. Used as an overload target to avoid side-effects on users who import
     Pytensor.

pytensor/link/numba/dispatch/linalg/solve/lu_solve.py

@@ -92,7 +92,7 @@ def _lu_solve(
     trans: int,
     overwrite_b: bool,
     check_finite: bool,
-) -> np.ndarray:
+):
     """
     Thin wrapper around scipy.lu_solve, used to avoid side effects from numba overloads on users who import Pytensor.
     """
@@ -108,7 +108,7 @@ def lu_solve_impl(
     trans: int,
     overwrite_b: bool,
     check_finite: bool,
-) -> Callable[[tuple[np.ndarray, np.ndarray], np.ndarray, bool, bool], np.ndarray]:
+) -> Callable[[np.ndarray, np.ndarray, np.ndarray, bool, bool, bool], np.ndarray]:
     ensure_lapack()
     _check_scipy_linalg_matrix(lu_and_piv[0], "lu_solve")
     _check_scipy_linalg_matrix(b, "lu_solve")

pytensor/npy_2_compat.py

@@ -41,7 +41,7 @@
 
 
 if using_numpy_2:
-    ndarray_c_version = np._core._multiarray_umath._get_ndarray_c_version()
+    ndarray_c_version = np._core._multiarray_umath._get_ndarray_c_version()  # type: ignore[attr-defined]
 else:
     ndarray_c_version = np.core._multiarray_umath._get_ndarray_c_version()  # type: ignore[attr-defined]