docs: def get_variables_in_module(module,
                              collection=tf.GraphKeys.TRAINABLE_VARIABLES):
    """Returns tuple of `tf.Variable`s declared inside an `snt.Module`.
  
    Note that this operates by searching the variable scope a module contains,
    and so does not know about any modules which were constructed elsewhere but
    used inside this module.
  
    Args:
      module: `snt.Module` instance to query the scope of.
      collection: Collection to restrict query to. By default this is
        `tf.Graphkeys.TRAINABLE_VARIABLES`, which doesn't include non-trainable
        variables such as moving averages.
  
    Returns:
      A tuple of `tf.Variable` objects.
  
    Raises:
      NotConnectedError: If the module is not connected to the Graph.
    """
    return module.get_variables(collection=collection)