def METHOD_NAME(self, y_gt, y_pr, sample_weight=None):
    return compute_ohkm_loss(
        y_gt,
        y_pr,
        hard_to_easy_ratio=self.hard_to_easy_ratio,
        min_hard_keypoints=self.min_hard_keypoints,
        max_hard_keypoints=self.max_hard_keypoints,
        loss_scale=self.loss_scale,
    )