code: def replace_node_module(
      node: fx.Node,
      modules: Dict[str, Any],
      new_module: torch.nn.Module,
  ):
      assert isinstance(node.target, str)
      parent_name, name = _parent_name(node.target)
      modules[node.target] = new_module
      setattr(modules[parent_name], name, new_module)
  
imports: import torch.fx
  from torch.fx.node import Argument, Target
  from torch.nn.utils.fusion import fuse_conv_bn_eval
  from typing import Type, Dict, Any, Tuple, Iterable, Optional, List, cast
  import torch
  import torch.nn
  import torch.nn.functional
  from torch.fx.passes.shape_prop import ShapeProp
  import copy
  from collections import defaultdict
  import torch.utils.mkldnn
  import operator
  import time
  import logging
  from enum import Enum
label_desc: Use image node_id 3 for calling a global function with example usage: replace_node_module(node, modules, new_module) without return types