diff --git a/tensorflow_probability/python/distributions/inverse_gaussian_test.py b/tensorflow_probability/python/distributions/inverse_gaussian_test.py index d07034f09d..3bf74c3307 100644 --- a/tensorflow_probability/python/distributions/inverse_gaussian_test.py +++ b/tensorflow_probability/python/distributions/inverse_gaussian_test.py @@ -383,6 +383,7 @@ def testInverseGaussianFullyReparameterized(self): @test_util.numpy_disable_gradient_test def testCompareToExplicitGradient(self): """Compare to the explicit reparameterization derivative.""" + self.skipTest('b/331471078') concentration_np = np.arange(4)[..., np.newaxis] + 1. concentration = tf.constant(concentration_np, self.dtype) loc_np = np.arange(3) + 1.