Further pylint/style fixes

marvinpfoertner · Jul 11, 2023 · f0a8da8 · f0a8da8
1 parent 8629250
commit f0a8da8
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 7 deletions.
diff --git a/src/linpde_gp/randprocs/covfuncs/linfunctls/_registry.py b/src/linpde_gp/randprocs/covfuncs/linfunctls/_registry.py
@@ -68,13 +68,12 @@ def _(self, k: covfuncs.StackCovarianceFunction, /, *, argnum: int = 0):
     from ...crosscov import (  # pylint: disable=import-outside-toplevel
         StackedProcessVectorCrossCovariance,
     )
+
     L_covfuncs = np.copy(k.covfuncs)
     for idx, covfunc in np.ndenumerate(L_covfuncs):
         L_covfuncs[idx] = self(covfunc, argnum=argnum)
 
-    return StackedProcessVectorCrossCovariance(
-        L_covfuncs
-    )
+    return StackedProcessVectorCrossCovariance(L_covfuncs)
 
 
 @linfunctls.LinearFunctional.__call__.register  # pylint: disable=no-member

diff --git a/src/linpde_gp/randprocs/crosscov/_stack.py b/src/linpde_gp/randprocs/crosscov/_stack.py
@@ -63,8 +63,8 @@ def _evaluate(self, x: np.ndarray) -> np.ndarray:
                 res[(..., *idx)] = eval_at_idx
         else:
             res = np.zeros(self._pv_crosscovs.shape + batch_shape + self.randvar_shape)
-            for idx, eval in np.ndenumerate(evals):
-                res[(*idx, ...)] = eval
+            for idx, eval_at_idx in np.ndenumerate(evals):
+                res[(*idx, ...)] = eval_at_idx
             # Move entire batch shape to the front
             res = np.moveaxis(
                 res,
@@ -86,8 +86,8 @@ def _evaluate_jax(self, x: jnp.ndarray) -> jnp.ndarray:
 
         if self.reverse:
             res = jnp.zeros(self.randvar_shape + batch_shape + self._pv_crosscovs.shape)
-            for idx, eval in np.ndenumerate(evals):
-                res.at[(..., *idx)].set(eval)
+            for idx, eval_at_idx in np.ndenumerate(evals):
+                res.at[(..., *idx)].set(eval_at_idx)
         else:
             res = jnp.zeros(self._pv_crosscovs.shape + batch_shape + self.randvar_shape)
             for idx, eval_at_idx in np.ndenumerate(evals):