Fix flaky test tolerance
@@ -28,7 +28,7 @@ class TestGridKernel(unittest.TestCase):
self.assertIsInstance(grid_covar, KroneckerProductLazyTensor)
grid_eval = kernel(grid_data, grid_data).evaluate()
actual_eval = base_kernel(grid_data, grid_data).evaluate()
- self.assertLess(torch.norm(grid_eval - actual_eval), 1.2e-5)
+ self.assertLess(torch.norm(grid_eval - actual_eval), 2e-5)
def test_nongrid_grid(self):
base_kernel = RBFKernel()
