code: def _module() -> tf.Module:
      class _Mod(tf.Module):
          def __init__(self) -> None:
              super().__init__()
              self.var = tf.Variable(0.0)
              self.param = Parameter(0.0)
  
      module = _Mod()
  
      assert len(module.trainable_variables) == 2
      assert len(module.variables) == 2
  
      return module
  
imports: import tensorflow
  from gpflow import Parameter, set_trainable
label_desc: Use image node_id 1 for calling a global function with example usage: _module() and returns: module