def _get_retrieved_documents(self, claim_id,
                               scrape_type):
    """Retrieve the appropriate set of documents depending on settings.

    Args:
      claim_id: The claim to get documents for
      scrape_type: The scrape type to use when fetching documents

    Returns:
      A list of documents to generate examples from
    """
    if scrape_type in (constants.DRQA, constants.LUCENE):
      claim_scrape = self._name_to_scrape[scrape_type][str(claim_id)]
      documents = [(doc.doc_id, doc.ir_score) for doc in claim_scrape.documents]
    elif scrape_type in constants.UKP_TYPES:
      if scrape_type == constants.UKP_WIKI:
        claim_scrape = self._ukp_docs[claim_id]['wiki_results']
      elif scrape_type == constants.UKP_PRED:
        claim_scrape = self._ukp_docs[claim_id]['predicted_pages']
      else:
        raise ValueError(f'Invalid scrape type: {scrape_type}')
      documents = [
          # UKP Does not have document scores.
          (doc_id.replace(' ', '_'), -1) for doc_id in claim_scrape
      ]
    else:
      raise ValueError(f'Invalid scrape type: {scrape_type}')
    return documents
Retrieve the appropriate set of documents depending on settings.

    Args:
      claim_id: The claim to get documents for
      scrape_type: The scrape type to use when fetching documents

    Returns:
      A list of documents to generate examples from
    
language/serene/fever_tfds.py