code: def _download_and_index_cqa(dl_manager):
      """Downloads CQA and returns it, indexed by id, for joining with Cos-E."""
  
      downloaded_files = dl_manager.download_and_extract(
          {
              "cqa_train": _CQA_URL_TRAIN,
              "cqa_dev": _CQA_URL_DEV,
              "cqa_test": _CQA_URL_TEST,
          }
      )
  
      # NB: "cqa_test" is included in the files, but not in any of the CoS-E splits.
      cqa_splits = ["cqa_train", "cqa_dev"]
      cqa_complete = []
      for split in cqa_splits:
          with tf.io.gfile.GFile(downloaded_files[split]) as f:
              for _, line in enumerate(f):
                  d = json.loads(line)
                  cqa_complete.append(d)
  
          # Index the CQA dataset by id for joining with Cos-E.
          cqa_indexed = {}
      for d in cqa_complete:
          cqa_indexed[d["id"]] = d
      return cqa_indexed
  
file_path: datasets/tensorflow_datasets/text/cos_e.py
imports: import json
  import os
  from etils import epath
  from tensorflow_datasets.core.utils.lazy_imports_utils import tensorflow
  import tensorflow_datasets.public_api
label_desc: Use image node_id 1 for calling a global function with example usage: _download_and_index_cqa(dl_manager) and returns: cqa_indexed