code: def _build_pruned_layer_model(self, layer):
      wrapped_layer = prune.prune_low_magnitude(layer, **self.params)
      self.model.add(wrapped_layer)
      self.model.build(input_shape=(10, 1))
  
      return wrapped_layer
  
imports: import json
  import tempfile
  import numpy
  import tensorflow
  from tensorflow_model_optimization.python.core.keras import test_utils
  from tensorflow_model_optimization.python.core.sparsity.keras import prunable_layer
  from tensorflow_model_optimization.python.core.sparsity.keras import prune
  from tensorflow_model_optimization.python.core.sparsity.keras import pruning_schedule
  from tensorflow_model_optimization.python.core.sparsity.keras import pruning_wrapper
label_desc: Use image node_id 2 for calling the PruneTest obj's underlying member method code with example usage: obj._build_pruned_layer_model(layer) and returns: wrapped_layer