code: def test_simple_adaptation(self):
      builder = kernel_builder.KernelBuilder.make(fake_target_log_prob)
      builder = builder.hmc(num_leapfrog_steps=3).simple_adaptation(
          adaptation_rate=0.03
      )
      outputs1 = builder.sample(10, tf.cast(0.5, tf.float32))
      self.assertIsNotNone(outputs1.new_step_size)
      builder = builder.set_step_size(
          outputs1.new_step_size
      ).clear_step_adapter()
      outputs2 = builder.sample(10, outputs1.current_state)
      diagnostics = outputs2.get_diagnostics()
      self.assertIn("realized_acceptance_rate", diagnostics)
  
imports: import tensorflow.compat.v2
  from tensorflow_probability.python.experimental.mcmc import kernel_builder
  from tensorflow_probability.python.internal import test_util
label_desc: Use image node_id 2 for calling the TestKernelOutputs obj's underlying member method code with example usage: obj.test_simple_adaptation() without return types