code: def testCollectPrunableLayers(self):
      lstm_layer = keras.layers.RNN(
          layers.LSTMCell(4, dropout=0.5, recurrent_dropout=0.5),
          input_shape=(None, 4),
      )
      self.model.add(Prune(lstm_layer))
      self.model.add(Prune(layers.BatchNormalization()))
      self.model.add(Prune(layers.Flatten()))
      self.model.add(Prune(layers.Dense(10)))
      self.model.add(Prune(layers.Dropout(0.5)))
  
      self.model.build(input_shape=(1, 4))
  
      self.assertLen(
          pruning_wrapper.collect_prunable_layers(self.model), 5
      )
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import tensorflow
  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 10 for calling the PruningWrapperTest obj's underlying member method code with example usage: obj.testCollectPrunableLayers() without return types