QuantizationConfig config{};
  CalibrationOptions& calibration_options =
      *config.mutable_calibration_options();
  calibration_options.set_calibration_method(
      CalibrationOptions::CALIBRATION_METHOD_HISTOGRAM_PERCENTILE);
  calibration_options.mutable_calibration_parameters()->set_num_bins(512);

  // Test that if the user explicitly provided part of the
  // `calibration_options`, it is not overridden, rest of the data are default.

          "calibration_options": stablehlo_quant_config_pb2.CalibrationOptions(
              calibration_method=_CalibrationMethod.CALIBRATION_METHOD_HISTOGRAM_MSE_MAX_FREQUENCY,
              calibration_parameters=stablehlo_quant_config_pb2.CalibrationOptions.CalibrationParameters(),
          ),
      },
      {
          "testcase_name": "with_histogram_mse_symmetric",
          "calibration_options": stablehlo_quant_config_pb2.CalibrationOptions(

        CalibrationOptions::CALIBRATION_METHOD_MIN_MAX);
  }

  switch (quant_config.calibration_options().calibration_method()) {
    case CalibrationOptions::CALIBRATION_METHOD_HISTOGRAM_PERCENTILE:
    case CalibrationOptions::CALIBRATION_METHOD_HISTOGRAM_MSE_BRUTEFORCE:
    case CalibrationOptions::CALIBRATION_METHOD_HISTOGRAM_MSE_MAX_FREQUENCY:
    case CalibrationOptions::CALIBRATION_METHOD_HISTOGRAM_MSE_SYMMETRIC:

    const CalibrationOptions& calibration_options,
    const PyFunctionLibrary& py_function_library) {
  TF_ASSIGN_OR_RETURN(const CalibrationStatisticsFlatMap statistics_map,
                      ReadStatistics(calibration_data_dir));

  absl::Status status = absl::OkStatus();
  module_op.walk([&py_function_library, &calibration_options, &status,

// The min and max values will be added to the "min" and "max" attributes,
// respectively. `calibration_options` provides the strategy to retrieve min and
// max values.
absl::Status AddCalibrationStatistics(
    mlir::ModuleOp module_op, absl::string_view calibration_data_dir,
    const stablehlo::quantization::CalibrationOptions& calibration_options,
    const tensorflow::quantization::PyFunctionLibrary& py_function_library);

  GetCalibrationMinMaxValue(const tensorflow::calibrator::CalibrationStatistics&
                                calibration_statistics,
                            const ::stablehlo::quantization::CalibrationOptions&
                                calibration_options) const = 0;
  // LINT.ThenChange(
  //     pywrap_function_lib.pyi:get_calibration_min_max_value,
  //     py_function_lib.py:get_calibration_min_max_value,
  // )
};

  std::string calibration_data_dir =
      config.calibration_options().calibration_data_dir();
  if (calibration_data_dir.empty()) {
    TF_ASSIGN_OR_RETURN(calibration_data_dir, CreateTmpDir());
  }

  TF_RETURN_IF_ERROR(ExportToSavedModel(
      module_op, calibration_data_dir,
      config.calibration_options().force_regenerate_calibration_data(),
      calibration_saved_model_dir));

// * StaticRangePtqPreset
//   - The preset's `representative_datasets` field will be transferred to
//   `QuantizationConfig.calibration_options.representative_datasets`, unless
//   the user explicitly provided representative dataset configs to
//   `calibration_options`. In that case, the explicit configs take precedence
//   and the preset's configs are ignored.

      const CalibrationStatistics& calibration_statistics,
      const CalibrationOptions& calibration_options) const override {
    PYBIND11_OVERRIDE_PURE(std::optional<MinMaxValue>, PyFunctionLibrary,
                           get_calibration_min_max_value,
                           calibration_statistics, calibration_options);
  }
};

}  // namespace

PYBIND11_MODULE(pywrap_function_lib, m) {

      },
      {
          'calibration_options': qc.CalibrationOptions(
              calibration_method=_CalibrationMethod.CALIBRATION_METHOD_AVERAGE_MIN_MAX
          ),
      },
      {
          'calibration_options': qc.CalibrationOptions(
              calibration_method=_CalibrationMethod.CALIBRATION_METHOD_HISTOGRAM_PERCENTILE,
              calibration_parameters=qc.CalibrationOptions.CalibrationParameters(