code: def krm_get_batch_context_only_knowledge(
      self, batch: Batch, orig_fun: Callable[[Batch], torch.Tensor]
  ):
      """
      Monkey-patch get_batch_context for the KRM to block on the prior knowledge
      responses.
  
      Discard conversational context.
      """
      ctxts = orig_fun(batch)
      if "prior_knowledge_responses_vec" in batch:
          ctxts = batch.prior_knowledge_responses_vec.to(ctxts)
      return ctxts
  
file_path: ParlAI/projects/seeker/utils.py
imports: from collections import Counter
  from enum import Enum
  import torch
  from typing import Callable, List, Tuple
  from parlai.core.torch_agent import Batch
  from parlai.tasks.wizard_of_wikipedia.agents import TOKEN_KNOWLEDGE, TOKEN_END_KNOWLEDGE
  import parlai.utils.logging
label_desc: Use image node_id 3 for calling a global function with example usage: krm_get_batch_context_only_knowledge(batch, orig_fun) and returns: ctxts