code: def testPruneTrainingRaisesError_PruningStepCallbackMissing(self):
      pruned_model, x_train, y_train = self._pruned_model_setup()
  
      # Throws an error since UpdatePruningStep is missing.
      with self.assertRaises(errors_impl.InvalidArgumentError):
          pruned_model.fit(x_train, y_train)
  
file_path: model-optimization/tensorflow_model_optimization/python/core/sparsity/keras/pruning_callbacks_test.py
imports: import os
  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 prune
  from tensorflow_model_optimization.python.core.sparsity.keras import pruning_callbacks
label_desc: Use image node_id 6 for calling the PruneCallbacksTest obj's underlying member method code with example usage: obj.testPruneTrainingRaisesError_PruningStepCallbackMissing() without return types