Skip to content

Commit ef6de40

Browse files
Fix matern tests
1 parent 2674a41 commit ef6de40

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

src/probnum/randprocs/kernels/_matern.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -94,12 +94,12 @@ def _evaluate(self, x0: ArrayType, x1: Optional[ArrayType] = None) -> ArrayType:
9494
return backend.exp(-1.0 / (2.0 * self.lengthscale ** 2) * distances ** 2)
9595

9696
# The modified Bessel function K_nu is not defined for z=0
97-
# distances = backend.maximum(distances, backend.finfo(distances.dtype).eps)
98-
99-
# scaled_distances = backend.sqrt(2 * self.nu) / self.lengthscale * distances
100-
# return (
101-
# 2 ** (1.0 - self.nu)
102-
# / backend.special.gamma(self.nu)
103-
# * scaled_distances ** self.nu
104-
# * backend.special.kv(self.nu, scaled_distances)
105-
# )
97+
distances = backend.maximum(distances, backend.finfo(distances.dtype).eps)
98+
99+
scaled_distances = backend.sqrt(2 * self.nu) / self.lengthscale * distances
100+
return (
101+
2 ** (1.0 - self.nu)
102+
/ backend.special.gamma(self.nu)
103+
* scaled_distances ** self.nu
104+
* backend.special.kv(self.nu, scaled_distances)
105+
)

0 commit comments

Comments
 (0)