code: def get_graph_node_number(graph):
      # FIXME: if the feature is None, this will throw an error
      if isinstance(graph, GeneralStaticGraph):
          if BACKEND == "pyg":
              return graph.nodes.data["x"].size(0)
          return graph.nodes.data["attr"].size(0)
      if BACKEND == "pyg":
          size = graph.x.shape[0]
      else:
          size = graph.num_nodes()
      return size
  
imports: import random
  import typing
  import torch
  import torch.backends.cudnn
  import numpy
  import pandas
  from ..backend import DependentBackend
  from ..data import Dataset
  from ..data.graph import GeneralStaticGraph
  from ..utils import get_logger
label_desc: Use image node_id 2 for calling a global function with example usage: get_graph_node_number(graph) and returns: size, graph, graph