diff --git a/tensorflow_probability/python/mcmc/hmc_test.py b/tensorflow_probability/python/mcmc/hmc_test.py index c2d4082710..13e398aff7 100644 --- a/tensorflow_probability/python/mcmc/hmc_test.py +++ b/tensorflow_probability/python/mcmc/hmc_test.py @@ -1064,7 +1064,8 @@ def unnormalized_posterior_log_prob(w): weights_prior_estimated_scale_[-5:].mean(), err=0.005) - @test_util.run_in_graph_and_eager_modes + # TODO(b/112427830): Reenable eager tests for TF 1.11 release. + @run_in_graph_mode_only() def test_step_size_adapts(self): dtype = np.float32