code: def build_mnist_model(model_type, pruning_params=None):
      return {
          "sequential": _build_mnist_sequential_model(),
          "functional": _build_mnist_functional_model(),
          "layer_list": _build_mnist_layer_list(),
          "layer_wise": _build_mnist_layerwise_pruned_model(
              pruning_params
          ),
      }[model_type]
  
file_path: model-optimization/tensorflow_model_optimization/python/core/sparsity/keras/test_utils.py
imports: import tempfile
  import numpy
  import tensorflow
  from tensorflow_model_optimization.python.core.sparsity.keras import prune
  from tensorflow_model_optimization.python.core.sparsity.keras import pruning_utils
  from tensorflow_model_optimization.python.core.sparsity.keras import pruning_wrapper
label_desc: Use image node_id 5 for calling a global function with example usage: build_mnist_model(model_type, pruning_params) and returns: dict