src_saved_model_path: str,
      tags: set[str],
      serialized_signature_def_map: dict[str, bytes],
  ) -> Optional[bool]: ...
  # LINT.ThenChange()

  # LINT.IfChange(run_calibration)
  def run_calibration(
      self,
      saved_model_path: str,
      signature_keys: list[str],
      tags: set[str],
      force_graph_mode_calibration: bool,
      # Value type: RepresentativeDatasetFile.

           py::arg("exported_model_serialized"),
           py::arg("src_saved_model_path"), py::arg("tags"),
           py::arg("serialized_signature_def_map"))
      .def("run_calibration", &PyFunctionLibrary::RunCalibration,
           py::arg("saved_model_path"), py::arg("signature_keys"),
           py::arg("tags"), py::arg("force_graph_mode_calibration"),
           py::arg("representative_dataset_file_map_serialized"))

  //
  // If the function signature changes, likely its corresponding .pyi type
  // hinting and definition should also change.
  // LINT.IfChange(run_calibration)
  virtual std::optional<bool> RunCalibration(
      absl::string_view saved_model_path,
      const std::vector<std::string>& signature_keys,
      const std::unordered_set<std::string>& tags,
      bool force_graph_mode_calibration,

  # file.
  # LINT.IfChange(run_calibration)
  def run_calibration(
      self,
      saved_model_path: str,
      signature_keys: list[str],
      tags: set[str],
      force_graph_mode_calibration: bool,
      representative_dataset_file_map_serialized: dict[str, bytes],
  ) -> Optional[bool]:
    # LINT.ThenChange(py_function_lib.h:run_calibration)

    TF_ASSIGN_OR_RETURN(
        const auto representative_dataset_file_map,
        CreateRepresentativeDatasetFileMap(dataset_config_vector));

    // Run calibration on the exported model.
    if (py_function_lib_->RunCalibration(
            calibration_saved_model_dir, signature_keys_, tags_,
            /*force_graph_mode_calibration=*/true,
            representative_dataset_file_map) == std::nullopt) {
      return absl::InternalError(

    py_function_library.SaveExportedModel(
        calibration_saved_model_dir, calibration_exported_model,
        saved_model_path, tags, signature_def_map);

    py_function_library.RunCalibration(
        calibration_saved_model_dir, signature_keys, tags,
        quantization_options.force_graph_mode_calibration(),
        representative_dataset_file_map_serialized);
  }