class_docstr: Override TGA to use a different tree search decoder.
docstr: Override to always provide full context.
docstr_tok: ["Override","to","always","provide","full","context","."]
code: def _get_batch_context(self, batch):
      """
      Override to always provide full context.
      """
      if "full_text_vec" not in batch:
          logging.warn(
              "Batch does not have full text vec, resorting to text vec"
          )
          return batch.text_vec
      return batch.full_text_vec
  
imports: import random
  import torch
  import torch.nn.functional
  from typing import Optional, Any, Dict, List
  from parlai.agents.rag.retrievers import clean_vec
  from parlai.agents.transformer.transformer import TransformerGeneratorAgent
  from parlai.core.opt import Opt
  from parlai.core.params import ParlaiParser
  from parlai.core.torch_generator_agent import TreeSearch, GreedySearch, BeamSearch, DelayedBeamSearch, TopKSampling, NucleusSampling, TSType
  import parlai.utils.logging
  from parlai.utils.torch import neginf
  from projects.light_whoami.agents.rpa_rerank import RPAReranker, RPARerankAgent, LongRPARerankAgent
  from projects.light_whoami.task.utils import extract_characters
  from projects.msc.agents.long_tga import TransformerVariantAgent
  from parlai.agents.reranker.reranker import AbstractReranker
label_desc: Use image node_id 5 for calling the PacerAgentMixin obj's underlying member method code with example usage: obj._get_batch_context(batch) and returns: batch, batch