def METHOD_NAME(
    node: flow.fx.Node, modules: Dict[str, Any], new_module: flow.nn.Module
):
    assert isinstance(node.target, str)
    parent_name, name = _parent_name(node.target)
    setattr(modules[parent_name], name, new_module)