code: def build_container_map(self) -> ContainerMap:
      map = ContainerMap()
  
      transformer_containers = [
          MistralTransformerContainer(self.model)
          for _ in range(self.model.num_layers)
      ]
  
      map.set_transformer_params(
          ["model.layers"], transformer_containers
      )
  
      map.set_non_transformer_params(
          MistralNonTransformerContainer(self.model)
      )
  
      map.set_unmapped_params([])
  
      return map
  
file_path: DeepSpeed/deepspeed/inference/v2/model_implementations/mistral/policy.py
imports: from typing import Any
  from ...config_v2 import RaggedInferenceEngineConfig
  from ..inference_policy_base import ContainerMap, InferenceV2Policy
  from .container import MistralNonTransformerContainer, MistralTransformerContainer
  from .model import MistralInferenceModel
label_desc: Use image node_id 2 for calling the MistralPolicy obj's underlying member method code with example usage: obj.build_container_map() and returns: map