code: def _trace_current_state(current_state, kernel_results):
      del kernel_results
      return current_state
  
file_path: probability/tensorflow_probability/python/experimental/mcmc/sample_fold.py
imports: import warnings
  import tensorflow.compat.v2
  from tensorflow_probability.python.experimental.mcmc import sample
  from tensorflow_probability.python.experimental.mcmc import sample_discarding_kernel
  from tensorflow_probability.python.experimental.mcmc import step
  from tensorflow_probability.python.experimental.mcmc import thinning_kernel
  from tensorflow_probability.python.experimental.mcmc import with_reductions
  from tensorflow_probability.python.internal import samplers
  from tensorflow.python.util import nest
label_desc: Use image node_id 2 for calling a global function with example usage: _trace_current_state(current_state, kernel_results) and returns: current_state