code: def testPruneCheckpoints_CheckpointsNotSparse(self):
      is_model_sparsity_not_list = []
  
      # Run multiple times since problem doesn't always happen.
      for _ in range(3):
          model = keras_test_utils.build_simple_dense_model()
          pruned_model = prune.prune_low_magnitude(model, **self.params)
  
          checkpoint_dir = tempfile.mkdtemp()
          checkpoint_path = checkpoint_dir + "/weights.{epoch:02d}.tf"
  
          callbacks = [
              pruning_callbacks.UpdatePruningStep(),
              tf.keras.callbacks.ModelCheckpoint(
                  filepath=checkpoint_path,
                  save_weights_only=True,
                  save_freq=1,
              ),
          ]
  
          # Train one step. Sparsity reaches final sparsity.
          self._train_model(pruned_model, epochs=1, callbacks=callbacks)
          test_utils.assert_model_sparsity(self, 0.5, pruned_model)
  
          latest_checkpoint = tf.train.latest_checkpoint(checkpoint_dir)
  
          same_architecture_model = (
              keras_test_utils.build_simple_dense_model()
          )
          pruned_model = prune.prune_low_magnitude(
              same_architecture_model, **self.params
          )
  
          # Sanity check.
          test_utils.assert_model_sparsity(self, 0, pruned_model)
  
          pruned_model.load_weights(latest_checkpoint)
          is_model_sparsity_not_list.append(
              test_utils.is_model_sparsity_not(0.5, pruned_model)
          )
  
      self.assertTrue(any(is_model_sparsity_not_list))
  
imports: import tempfile
  from absl.testing import parameterized
  import numpy
  import tensorflow
  from tensorflow_model_optimization.python.core.keras import test_utils
  from tensorflow_model_optimization.python.core.sparsity.keras import prune
  from tensorflow_model_optimization.python.core.sparsity.keras import prune_registry
  from tensorflow_model_optimization.python.core.sparsity.keras import pruning_callbacks
  from tensorflow_model_optimization.python.core.sparsity.keras import pruning_schedule
  from tensorflow_model_optimization.python.core.sparsity.keras import pruning_wrapper
  from tensorflow_model_optimization.python.core.sparsity.keras import test_utils
label_desc: Use image node_id 26 for calling the PruneIntegrationTest obj's underlying member method code with example usage: obj.testPruneCheckpoints_CheckpointsNotSparse() without return types