Relax tolerance for test_gaussian_process_log_prob_gradient

PiperOrigin-RevId: 619566060
tensorflow · Mar 27, 2024 · 7e1ce86 · 7e1ce86
1 parent c9b3ac5
commit 7e1ce86
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensorflow_probability/python/experimental/fastgp/fast_gp_test.py b/tensorflow_probability/python/experimental/fastgp/fast_gp_test.py
@@ -458,7 +458,7 @@ def slow_log_prob(amplitude, length_scale, noise, jitter):
     self.assertAlmostEqual(value, slow_value, delta=8)
     slow_d_amp, slow_d_length_scale, slow_d_noise, slow_d_jitter = slow_gradient
     self.assertAlmostEqual(d_amp, slow_d_amp, delta=0.01)
-    self.assertAlmostEqual(d_length_scale, slow_d_length_scale, delta=0.001)
+    self.assertAlmostEqual(d_length_scale, slow_d_length_scale, delta=0.002)
     # TODO(thomaswc): Investigate why the noise gradient is so noisy.
     self.assertAlmostEqual(d_noise, slow_d_noise, delta=0.5)
     # TODO(thomaswc): Investigate why slow_d_jitter is zero.