def METHOD_NAME(self, reduction, p, q, log_prob):
    """Test the differentiability of the metric, according to its `is_differentiable` attribute."""
    self.run_differentiability_test(
        p,
        q,
        metric_module=KLDivergence,
        metric_functional=kl_divergence,
        metric_args={"log_prob": log_prob, "reduction": reduction},
    )