From 51f1ed39ebd6152ffad8bf91040e37293dc8dfad Mon Sep 17 00:00:00 2001 From: ursk Date: Mon, 1 Apr 2024 09:30:43 -0700 Subject: [PATCH] Relax tolerances for psd_kernel_properties_test PiperOrigin-RevId: 620869405 --- .../python/math/psd_kernels/psd_kernel_properties_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow_probability/python/math/psd_kernels/psd_kernel_properties_test.py b/tensorflow_probability/python/math/psd_kernels/psd_kernel_properties_test.py index f57f61d0b5..13be38c493 100644 --- a/tensorflow_probability/python/math/psd_kernels/psd_kernel_properties_test.py +++ b/tensorflow_probability/python/math/psd_kernels/psd_kernel_properties_test.py @@ -220,7 +220,7 @@ def _test_slicing( # Check that applying a sliced kernel produces the same results as slicing # the results from the original. - self.assertAllClose(results[apply_slices], sliced_results, rtol=1e-5) + self.assertAllClose(results[apply_slices], sliced_results, rtol=2e-5) @parameterized.named_parameters( {'testcase_name': dname, 'kernel_name': dname}