message: Add call to compile to force the use of Keras v1 execution path in legacy contrib code.
diff: @@ -160,6 +160,9 @@ class AttentionMechanismTest(test.TestCase, parameterized.TestCase):
  config, custom_objects={attention_cls.__name__: attention_cls})
  loaded_model.set_weights(weights)
  + # TODO(b/138592586): Run with single-execution-path
  + loaded_model.compile("rmsprop", "mse", experimental_run_tf_function=False)
  +
  y = loaded_model.predict_on_batch([x_test, self.query, self.state])
  self.assertAllClose(y_ref, y)
  