code: def __init__(self):
      super(TestModel, self).__init__(
          default_event_space_bijector=tfb.Exp(),
          event_shape=[],
          dtype=tf.float32,
          name="test_model",
          pretty_name="TestModel",
          sample_transformations=dict(
              identity=targets.Model.SampleTransformation(
                  fn=lambda x: x,
                  pretty_name="Identity",
                  ground_truth_mean=1.0,  # It's wrong, but we'll use it for
                  # testing.
              ),
          ),
      )
  
file_path: probability/spinoffs/inference_gym/inference_gym/targets/model_test.py
imports: import numpy
  import tensorflow.compat.v2
  import tensorflow_probability
  from tensorflow_probability.python.internal import test_util
  from inference_gym import targets
  from inference_gym.internal import test_util
  from tensorflow.python.util import nest
label_desc: Use image node_id 1 to create a new TestModel object from inherited base classes: targets with example: obj = TestModel()