diff --git a/tensorflow_probability/python/distributions/bates_test.py b/tensorflow_probability/python/distributions/bates_test.py index d252a675a6..f2ca52b216 100644 --- a/tensorflow_probability/python/distributions/bates_test.py +++ b/tensorflow_probability/python/distributions/bates_test.py @@ -257,7 +257,7 @@ def testBatesPDFisNormalized(self, total_count, bounds): x = tf.linspace(low, high, nx) y = self.evaluate(d.prob(x)) dx = self.evaluate(x[1] - x[0]) - self.assertAllClose(scipy.integrate.simps(y=y, dx=dx), 1., + self.assertAllClose(scipy.integrate.simpson(y=y, dx=dx), 1., atol=5e-05, rtol=5e-05) def testBatesPDFonNaNs(self):