Skip to content

Commit ba3e3d1

Browse files
committed
Add suggestions from code review
1 parent 5f5ba7b commit ba3e3d1

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/probnum/randprocs/covfuncs/_arithmetic_fallbacks.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99
import numpy as np
1010

1111
from probnum import linops, utils
12-
from probnum.typing import NotImplementedType, ScalarLike
12+
from probnum.typing import ArrayLike, NotImplementedType, ScalarLike
1313

1414
from ._covariance_function import BinaryOperandType, CovarianceFunction
1515

@@ -56,7 +56,7 @@ def _evaluate(self, x0: np.ndarray, x1: Optional[np.ndarray] = None) -> np.ndarr
5656
return self._scalar * self._covfunc(x0, x1)
5757

5858
def linop(
59-
self, x0: utils.ArrayLike, x1: Optional[utils.ArrayLike] = None
59+
self, x0: ArrayLike, x1: Optional[ArrayLike] = None
6060
) -> linops.LinearOperator:
6161
return self._scalar * self._covfunc.linop(x0, x1)
6262

@@ -104,7 +104,7 @@ def _evaluate(self, x0: np.ndarray, x1: Optional[np.ndarray]) -> np.ndarray:
104104
)
105105

106106
def linop(
107-
self, x0: utils.ArrayLike, x1: Optional[utils.ArrayLike] = None
107+
self, x0: ArrayLike, x1: Optional[ArrayLike] = None
108108
) -> linops.LinearOperator:
109109
return functools.reduce(
110110
operator.add, (summand.linop(x0, x1) for summand in self._summands)

0 commit comments

Comments
 (0)