code: def _unflatten_model(components, structure_with_callables):
      model_components = []
      i = 0
      for c in tf.nest.flatten(structure_with_callables):
          if c is None:
              model_components.append(components["model"][i])
              i += 1
          else:
              model_components.append(c)
      return tf.nest.pack_sequence_as(
          structure_with_callables, model_components
      )
  
imports: import collections
  import functools
  import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import joint_distribution_sequential
  from tensorflow_probability.python.internal import auto_composite_tensor
  from tensorflow_probability.python.internal import distribution_util
label_desc: Use image node_id 6 for calling a global function with example usage: _unflatten_model(components, structure_with_callables) and returns: tf