code: def assign_simple_node_features(ndata, g, ntype, assign_id=False):
      """
      Copies data to the given block from the corresponding nodes in the original graph.
      """
      for col in g.nodes[ntype].data.keys():
          if not assign_id and col == dgl.NID:
              continue
          induced_nodes = ndata[dgl.NID]
          ndata[col] = g.nodes[ntype].data[col][induced_nodes]
  
imports: import argparse
  import pickle
  import time
  import dgl
  import dgl.function
  import numpy
  import torch
  import torch.nn
  import torch.nn.functional
  from torch.utils.data import DataLoader, IterableDataset
  from ..None import utils
label_desc: Use image node_id 3 for calling a global function with example usage: assign_simple_node_features(ndata, g, ntype, assign_id) without return types