def _update_score_distance_offset(model_config, offset):
  """Updates the keypoint candidate selection metric. See CenterNet proto."""
  meta_architecture = model_config.WhichOneof("model")
  if meta_architecture == "center_net":
    if len(model_config.center_net.keypoint_estimation_task) == 1: