code: def testExampleDoc2(self):
      samples = np.ones(10)
      model = TestModel()
      for (
          _,
          sample_transformation,
      ) in model.sample_transformations.items():
          transformed_samples = sample_transformation(samples)
          square_diff = tf.nest.map_structure(
              lambda gtm, sm: (gtm - tf.reduce_mean(sm, axis=0)) ** 2,
              sample_transformation.ground_truth_mean,
              transformed_samples,
          )
          self.assertLess(self.evaluate(square_diff), 1e-3)
  
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 2 for calling the ModelTest obj's underlying member method code with example usage: obj.testExampleDoc2() without return types