Fix mypy errors

Author: michaelosthege

pymc/gp/hsgp_approx.py

@@ -189,7 +189,9 @@ def __init__(
         self._drop_first = drop_first
         self._m = m
         self._m_star = int(np.prod(self._m))
-        self._L = L
+        self._L: Optional[pt.TensorVariable] = None
+        if L is not None:
+            self._L = pt.as_tensor(L)
         self._c = c
 
         super().__init__(mean_func=mean_func, cov_func=cov_func)
@@ -198,13 +200,13 @@ def __add__(self, other):
         raise NotImplementedError("Additive HSGPs aren't supported.")
 
     @property
-    def L(self):
+    def L(self) -> pt.TensorVariable:
         if self._L is None:
             raise RuntimeError("Boundaries `L` required but still unset.")
         return self._L
 
     @L.setter
-    def L(self, value):
+    def L(self, value: TensorLike):
         self._L = pt.as_tensor_variable(value)
 
     def prior_linearized(self, Xs: TensorLike):
@@ -290,9 +292,7 @@ def prior_linearized(self, Xs: TensorLike):
         # If not provided, use Xs and c to set L
         if self._L is None:
             assert isinstance(self._c, (numbers.Real, np.ndarray, pt.TensorVariable))
-            self.L = set_boundary(Xs, self._c)
-        else:
-            self.L = self._L
+            self._L = pt.as_tensor(set_boundary(Xs, self._c))
 
         eigvals = calc_eigenvalues(self.L, self._m, tl=pt)
         phi = calc_eigenvectors(Xs, self.L, eigvals, self._m, tl=pt)

scripts/run_mypy.py

@@ -105,7 +105,6 @@ def check_no_unexpected_results(mypy_lines: Iterator[str]):
     Exits the process with non-zero exit code upon unexpected results.
     """
     df = mypy_to_pandas(mypy_lines)
-
     all_files = {
         str(fp).replace(str(DP_ROOT), "").strip(os.sep).replace(os.sep, "/")
         for fp in DP_ROOT.glob("pymc/**/*.py")