code: def get_save_name(column: Text) -> Text:
      """Save file names for the different columns, according to OGB."""
      if column == "labels":
          return "graph-label.csv.gz"
      if column == "num_nodes":
          return "num-node-list.csv.gz"
      if column == "node_feat":
          return "node-feat.csv.gz"
      if column == "num_edges":
          return "num-edge-list.csv.gz"
      if column == "edge_feat":
          return "edge-feat.csv.gz"
      if column == "edge_index":
          return "edge.csv.gz"
  
      raise ValueError("Invalid column name.")
  
file_path: datasets/tensorflow_datasets/datasets/ogbg_molpcba/dummy_data_generation.py
imports: import collections
  import itertools
  import random
  from typing import Dict, Iterable, List, Text
  from absl import app
  from absl import flags
  from etils import epath
  import numpy
  from tensorflow_datasets.core import utils
  from tensorflow_datasets.core.utils import resource_utils
  from tensorflow_datasets.core.utils.lazy_imports_utils import pandas
label_desc: Use image node_id 5 for calling a global function with example usage: get_save_name(column) and returns: str, str, str, str, str, str