code: def configure_model_integ_grads_embeddings(model):
      r"""
      Wraps Pytext's DocNN model embedding with `IntegratedGradientsEmbedding`
      IntegratedGradientsEmbedding allows to perform baseline related operations
  
      Args
  
          model: a reference to DocModel
  
      Returns
  
          integrated_gradients_embedding_lst: The embedding layer which contains
                      IntegratedGradientsEmbedding as a wrapper over the original
                      embeddings of the model
  
      """
      embeddings = model.embedding
      integrated_gradients_embedding = PyTextInterpretableEmbedding(
          embeddings
      )
      return EmbeddingList([integrated_gradients_embedding], False)
  
imports: from collections import defaultdict
  import torch
  from pytext.models.embeddings.dict_embedding import DictEmbedding
  from pytext.models.embeddings.word_embedding import WordEmbedding
  from pytext.models.model import EmbeddingBase, EmbeddingList
label_desc: Use image node_id 2 for calling a global function with example usage: configure_model_integ_grads_embeddings(model) and returns: EmbeddingList