code: def test_dict_arg(self):
      @auto_composite_tensor.auto_composite_tensor
      class Test(auto_composite_tensor.AutoCompositeTensor):
          def __init__(self, var):
              self._var = var
  
          @property
          def var(self):
              return self._var
  
      self.assertAllClose(
          1.0,
          tf.function(lambda: Test(var={"a": tf.constant(1.0)}))().var[
              "a"
          ],
      )
  
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 16 for calling the AutoCompositeTensorTest obj's underlying member method code with example usage: obj.test_dict_arg() and returns: self