code: def test_works_correctly(self):
      if not tf.executing_eagerly():
          self.skipTest(
              "Skipping graph mode test since we simulate user behavior."
          )
      make_affine = functools.partial(
          tfn.AffineVariationalReparameterization,
          kernel_initializer=tfn.initializers.he_uniform(),
          bias_initializer=tfn.initializers.he_uniform(),
      )
      self.run_bnn_test(make_affine)
  
imports: import functools
  from absl.testing import parameterized
  import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.bijectors import softmax_centered
  from tensorflow_probability.python.bijectors import softplus
  from tensorflow_probability.python.distributions import independent
  from tensorflow_probability.python.distributions import normal
  from tensorflow_probability.python.experimental import nn
  from tensorflow_probability.python.internal import prefer_static
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.internal import tf_keras
  from tensorflow_probability.python.util import deferred_tensor
label_desc: Use image node_id 1 for calling the AffineVariationalReparameterizationTest obj's underlying member method code with example usage: obj.test_works_correctly() without return types