Up the tolerance for PSD slicing properties test.

PiperOrigin-RevId: 565146007
tensorflow · Sep 13, 2023 · 092bc6c · 092bc6c
1 parent c7e8b59
commit 092bc6c
Showing 1 changed file with 3 additions and 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
@@ -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}