code: def tearDownModule():
      # If `FLAGS.model_output_path` is set, serialize a `Model` instance to disk.
      # To update the serialized data read by `test_saved_model_from_disk`, pass
      # the local path to
      # `tensorflow_probability/python/internal/testdata/auto_composite_tensor`.
      # You may need to pass `--test_strategy=local` to avoid permissions errors.
      if FLAGS.model_output_path is not None:
          model = Model()
          tf.saved_model.save(model, FLAGS.model_output_path)
  
imports: import functools
  import os
  from absl import flags
  from absl.testing import parameterized
  import tensorflow.compat.v2
  from tensorflow_probability.python.bijectors import bijector
  from tensorflow_probability.python.bijectors import reshape
  from tensorflow_probability.python.bijectors import scale
  from tensorflow_probability.python.bijectors import shift
  from tensorflow_probability.python.bijectors import transform_diagonal
  from tensorflow_probability.python.distributions import independent
  from tensorflow_probability.python.distributions import normal
  from tensorflow_probability.python.distributions import transformed_distribution
  from tensorflow_probability.python.experimental.distributions import mvn_precision_factor_linop
  from tensorflow_probability.python.internal import auto_composite_tensor
  from tensorflow_probability.python.internal import tensor_util
  from tensorflow_probability.python.internal import test_util
  from absl.testing import absltest
label_desc: Use image node_id 1 for calling a global function with example usage: tearDownModule() without return types