representative_dataset_file_map);
  }

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

#include <utility>

namespace stablehlo::quantization {

// Represents the (min, max) value pair, representing the range of values after
// calibrating for quantization.
using MinMaxValue = std::pair<float, float>;

}  // namespace stablehlo::quantization

          absl::StrFormat("Calibrated data does not exist. Cannot find "
                          "statistics. value for id: %s",
                          id));
      return;
    }

    const std::optional<MinMaxValue> min_max_values =
        py_function_library.GetCalibrationMinMaxValue(iter->second,
                                                      calibration_options);
    if (min_max_values == std::nullopt) {

  // If the function signature changes, likely its corresponding .pyi type
  // hinting and definition should also change.
  // LINT.IfChange(get_calibration_min_max_value)
  virtual std::optional<stablehlo::quantization::MinMaxValue>
  GetCalibrationMinMaxValue(const tensorflow::calibrator::CalibrationStatistics&
                                calibration_statistics,
                            const ::stablehlo::quantization::CalibrationOptions&