code: def test_layer_adapter_custom_method(self):
      module = ModuleWithCustomForward()
  
      inputs = tf.keras.Input([], batch_size=1)
      layer = LayerAdapter(module=module, method="forward")
      output = layer(inputs)
      model = tf.keras.Model(inputs=inputs, outputs=output)
  
      self.assertEqual(model(tf.ones([])).numpy(), [2.0])
      self.assertEqual(model.trainable_variables, [module.w])
  
imports: from absl.testing import parameterized
  import sonnet
  from sonnet.src import test_utils
  from sonnet.src.conformance import descriptors
  import tensorflow
  import tree
label_desc: Use image node_id 6 for calling the KerasTest obj's underlying member method code with example usage: obj.test_layer_adapter_custom_method() without return types