code: def format_state_key(cls, key):
      return key.replace("fa_history", "fa_context")
  
imports: import copy
  import omegaconf
  import torch
  from mmf.common.registry import registry
  from mmf.models.base_model import BaseModel
  from mmf.modules.embeddings import ImageFeatureEmbedding, MultiHeadImageFeatureEmbedding, PreExtractedEmbedding, TextEmbedding
  from mmf.modules.layers import ClassifierLayer, ModalCombineLayer
  from mmf.utils.build import build_image_encoder
  from torch import nn
label_desc: Use image node_id 3 for calling the Pythia obj's underlying member method code with example usage: obj.format_state_key(cls, key) and returns: key