code: def setUp(self):
      super(PruneTest, self).setUp()
  
      # Layers passed in for Pruning can either be standard Keras layers provided
      # by the tf.keras API (these fall under the `keras.layers` namespace), or
      # custom layers provided by the user which inherit the base
      # `keras.layers.Layer`.
      # Standard Keras layers can either be Prunable (we know how to prune them),
      # Non-Prunable (we know these layers can't be pruned) and Unsupported (we
      # don't know how to deal with these yet.). Unsupported layers will raise an
      # error when tried to prune.
      # Custom Layers can either be Prunable (ie., they implement the
      # `PrunableLayer` interface, or Non-Prunable (they don't expose any pruning
      # behavior.)
  
      # TODO(pulkitb): Change naming of Prunable/Non-Prunable/Unsupported to be
      # clearer.
      self.keras_prunable_layer = layers.Dense(10)  # Supports pruning
      self.keras_non_prunable_layer = layers.Dropout(
          0.4
      )  # Pruning not applicable
      self.keras_unsupported_layer = layers.ConvLSTM2D(
          2, (5, 5)
      )  # Unsupported
      self.custom_prunable_layer = CustomPrunableLayer(10)
      self.custom_non_prunable_layer = CustomNonPrunableLayer(10)
  
      self.model = keras.Sequential()
      self.params = {
          "pruning_schedule": pruning_schedule.ConstantSparsity(0.5, 0),
          "block_size": (1, 1),
          "block_pooling_type": "AVG",
      }
  
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 1 for calling the PruneTest obj's underlying member method code with example usage: obj.setUp() without return types