pre-commit and mypy

Author: jessegrabowski

pytensor/link/numba/dispatch/slinalg.py

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

tests/tensor/test_elemwise.py

@@ -11,7 +11,7 @@
 import pytensor.scalar as ps
 import pytensor.tensor as pt
 import tests.unittest_tools as utt
-from pytensor import grad, In, Out
+from pytensor import In, Out, grad
 from pytensor.compile.function import function
 from pytensor.compile.mode import Mode
 from pytensor.configdefaults import config