code: def testClearData(self):
      with self.session():
          pywrap_calibration.clear_calibrator()
          input_tensor1 = array_ops.constant(
              [1.0, 2.0, 3.0, 4.0, 5.0], dtypes.float32
          )
          aggregator1 = custom_aggregator_op_wrapper.custom_aggregator(
              input_tensor1,
              "4",
              calibration_method=_CalibrationMethod.CALIBRATION_METHOD_MIN_MAX,
          )
          self.assertAllEqual(
              self.evaluate(aggregator1), [1.0, 2.0, 3.0, 4.0, 5.0]
          )
          input_tensor2 = array_ops.constant(
              [-1.0, -2.0, -3.0, -4.0, -5.0], dtypes.float32
          )
          aggregator2 = custom_aggregator_op_wrapper.custom_aggregator(
              input_tensor2,
              "5",
              calibration_method=_CalibrationMethod.CALIBRATION_METHOD_MIN_MAX,
          )
          self.assertAllEqual(
              self.evaluate(aggregator2), [-1.0, -2.0, -3.0, -4.0, -5.0]
          )
  
          statistics: calib_stat_pb2 = (
              pywrap_calibration.get_statistics_from_calibrator("4")
          )
          min_val = statistics.min_max_statistics.global_min
          max_val = statistics.min_max_statistics.global_max
          self.assertAllEqual((min_val, max_val), (1.0, 5.0))
  
          statistics: calib_stat_pb2 = (
              pywrap_calibration.get_statistics_from_calibrator("5")
          )
          min_val = statistics.min_max_statistics.global_min
          max_val = statistics.min_max_statistics.global_max
          self.assertAllEqual((min_val, max_val), (-5.0, -1.0))
  
          pywrap_calibration.clear_data_from_calibrator("4")
          with self.assertRaises(ValueError):
              pywrap_calibration.get_statistics_from_calibrator("4")
  
          statistics: calib_stat_pb2 = (
              pywrap_calibration.get_statistics_from_calibrator("5")
          )
          min_val = statistics.min_max_statistics.global_min
          max_val = statistics.min_max_statistics.global_max
          self.assertAllEqual((min_val, max_val), (-5.0, -1.0))
  
file_path: tensorflow-upstream/tensorflow/compiler/mlir/quantization/tensorflow/calibrator/integration_test/custom_aggregator_op_test.py
imports: import tensorflow
  from tensorflow.compiler.mlir.quantization.tensorflow import quantization_options_pb2
  from tensorflow.compiler.mlir.quantization.tensorflow.calibrator import calibration_statistics_pb2
  from tensorflow.compiler.mlir.quantization.tensorflow.calibrator import custom_aggregator_op_wrapper
  from tensorflow.compiler.mlir.quantization.tensorflow.calibrator import pywrap_calibration
  from tensorflow.python import pywrap_tensorflow
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import ops
  from tensorflow.python.ops import array_ops
  from tensorflow.python.platform import test
label_desc: Use image node_id 4 for calling the CustomAggregatorTest obj's underlying member method code with example usage: obj.testClearData() without return types