Skip to content

Commit 50e803f

Browse files
committed
add more decimal places to unit test
1 parent f62de39 commit 50e803f

File tree

1 file changed

+7
-8
lines changed

1 file changed

+7
-8
lines changed

test/kernels/test_matern52_kernel_grad.py

Lines changed: 7 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -20,14 +20,13 @@ def test_kernel(self, cuda=False):
2020
b = torch.tensor([[[1, 3], [0, 4]]], dtype=torch.float)
2121

2222
actual = torch.tensor(
23-
[
24-
[0.3056, -0.0000, 0.5822, 0.0188, -0.0210, 0.0420],
25-
[0.0000, 0.5822, 0.0000, 0.0210, -0.0056, 0.0532],
26-
[-0.5822, 0.0000, -0.8516, -0.0420, 0.0532, -0.0854],
27-
[0.1305, -0.2014, -0.2014, 0.0336, -0.0816, -0.0000],
28-
[0.2014, -0.1754, -0.3769, 0.0816, -0.1870, -0.0000],
29-
[0.2014, -0.3769, -0.1754, 0.0000, -0.0000, 0.0408],
30-
],
23+
[[ 0.3056225, -0.0000000, 0.5822443, 0.0188260, -0.0209871, 0.0419742],
24+
[ 0.0000000, 0.5822443, 0.0000000, 0.0209871, -0.0056045, 0.0531832],
25+
[-0.5822443, 0.0000000, -0.8515886, -0.0419742, 0.0531832, -0.0853792],
26+
[ 0.1304891, -0.2014212, -0.2014212, 0.0336440, -0.0815567, -0.0000000],
27+
[ 0.2014212, -0.1754366, -0.3768578, 0.0815567, -0.1870145, -0.0000000],
28+
[ 0.2014212, -0.3768578, -0.1754366, 0.0000000, -0.0000000, 0.0407784]
29+
]
3130
)
3231

3332
kernel = Matern52KernelGrad()

0 commit comments

Comments
 (0)