From eaa50d040b393f33def26053d9af16426e124558 Mon Sep 17 00:00:00 2001 From: Till Hoffmann Date: Fri, 16 Dec 2022 12:11:32 -0500 Subject: [PATCH] Update nugget variance for `rfft2` tests. --- gptools-util/tests/test_fft.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/gptools-util/tests/test_fft.py b/gptools-util/tests/test_fft.py index 1271dd2..6b5e987 100644 --- a/gptools-util/tests/test_fft.py +++ b/gptools-util/tests/test_fft.py @@ -68,7 +68,7 @@ def test_evaluate_log_prob_rfft2(batch_shape: tuple[int], rfft2_shape: int, use_ kernel = kernels.ExpQuadKernel(np.random.gamma(10, 0.1), np.random.gamma(10, 0.01), 1) \ + kernels.DiagonalKernel(1e-2, 1) cov = kernel.evaluate(xs) - np.fill_diagonal(cov, np.diag(cov) + 1e-9) + np.fill_diagonal(cov, np.diag(cov) + 1e-6) loc = np.random.normal(0, 1, xs.shape[0]) dist = stats.multivariate_normal(loc, cov) y = dist.rvs(batch_shape) @@ -80,7 +80,7 @@ def test_evaluate_log_prob_rfft2(batch_shape: tuple[int], rfft2_shape: int, use_ y2 = th.as_tensor(y2) loc2 = th.as_tensor(loc2) cov2 = th.as_tensor(cov2) - log_prob_rfft2 = fft.evaluate_log_prob_rfft2(y2, loc2, cov=cov2) + 1e-9 + log_prob_rfft2 = fft.evaluate_log_prob_rfft2(y2, loc2, cov=cov2) np.testing.assert_allclose(log_prob, log_prob_rfft2)