def _update_candidate_search_scale(model_config, search_scale):
  """Updates the keypoint candidate search scale in CenterNet."""
  meta_architecture = model_config.WhichOneof("model")
  if meta_architecture == "center_net":
    if len(model_config.center_net.keypoint_estimation_task) == 1: