code: def arrange_resource(wd_mentions_json):
      document_tokens_dict = dict()
      for mention_json in wd_mentions_json:
          mention_data = MentionData.read_json_mention_data_line(
              mention_json
          )
          mention_tokens = mention_data.tokens_number
          for i in range(0, len(mention_tokens)):
              doc_id = mention_data.doc_id
              sent_id = mention_data.sent_id
              token_map_key = MentionData.static_gen_token_unique_id(
                  doc_id, sent_id, mention_tokens[i]
              )
              document_tokens_dict[
                  token_map_key
              ] = mention_data.coref_chain
      return document_tokens_dict
  
file_path: nlp-architect/nlp_architect/data/cdc_resources/relations/within_doc_coref_extraction.py
imports: import logging
  import os
  from typing import List, Set
  from nlp_architect.common.cdc.mention_data import MentionData
  from nlp_architect.data.cdc_resources.relations.relation_extraction import RelationExtraction
  from nlp_architect.data.cdc_resources.relations.relation_types_enums import RelationType
  from nlp_architect.utils.io import load_json_file
label_desc: Use image node_id 2 for calling the WithinDocCoref obj's underlying member method code with example usage: obj.arrange_resource(wd_mentions_json) and returns: document_tokens_dict