From 092bc6ca805343f10d06b0a1d56ee9fae0182b1a Mon Sep 17 00:00:00 2001 From: siege Date: Wed, 13 Sep 2023 13:35:09 -0700 Subject: [PATCH] Up the tolerance for PSD slicing properties test. PiperOrigin-RevId: 565146007 --- .../python/math/psd_kernels/psd_kernel_properties_test.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) 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 5c9ee5b8bd..f57f61d0b5 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 @@ -218,9 +218,9 @@ def _test_slicing( (slices,)) apply_slices += tuple([slice(None)] * example_ndims) - # Check that sampling a sliced kernel produces the same shape as - # slicing the samples from the original. - self.assertAllClose(results[apply_slices], sliced_results, rtol=3e-6) + # 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) @parameterized.named_parameters( {'testcase_name': dname, 'kernel_name': dname}