From af26bd04194059d59d4870a0e4b9932fbc0329d4 Mon Sep 17 00:00:00 2001
From: "Joshua V. Dillon" <jvdillon@google.com>
Date: Thu, 9 Aug 2018 13:16:47 -0700
Subject: [PATCH] Disable MCEM HMC eager tests until TFP 0.3 release is cut.
 (TF 1.11 will have necessary fixes for tf.Variable and interaction with
 TensorArray to permit re-enabling eager mode tests.)

PiperOrigin-RevId: 208099686
---
 tensorflow_probability/python/mcmc/hmc_test.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

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