code: def get_model_and_pruner(mnist_model, pruner_cls, pruner_config):
      model = mnist_model
      pruner = pruner_cls(model, pruner_config)
      pruner.prepare(inplace=True)
      return model, pruner
  
file_path: coremltools/coremltools/test/optimize/torch/pruning/pruning_utils.py
imports: import numpy
  import torch
  import torch.nn.functional
  import os
label_desc: Use image node_id 5 for calling a global function with example usage: get_model_and_pruner(mnist_model, pruner_cls, pruner_config) and returns: model,  pruner