code: def __init__(self, nodes_list: List[Node]) -> None:
      self._node_list = nodes_list
      self._node_dict = {}
      self._set_node_dict()
  
imports: from typing import Any, Dict, List, Union
  from torch.fx.node import Node
  from colossalai.logging import get_dist_logger
label_desc: Use image node_id 1 to create a new NodeMgr object from inherited base classes: object with example: obj = NodeMgr(nodes_list)