code: def hmc_like_step_size_setter_fn(kernel_results, new_step_size):
      """Setter for `step_size` so it can be adapted."""
      return unnest.replace_innermost(
          kernel_results, step_size=new_step_size
      )
  
imports: import collections
  import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.internal import assert_util
  from tensorflow_probability.python.internal import dtype_util
  from tensorflow_probability.python.internal import prefer_static
  from tensorflow_probability.python.internal import tensorshape_util
  from tensorflow_probability.python.internal import unnest
  from tensorflow_probability.python.math.generic import reduce_logmeanexp
  from tensorflow_probability.python.mcmc import kernel
  from tensorflow_probability.python.mcmc.internal import util
label_desc: Use image node_id 1 for calling a global function with example usage: hmc_like_step_size_setter_fn(kernel_results, new_step_size) and returns: unnest