def METHOD_NAME(self) -> Tensor:
    """Compute normalized mutual information over state."""
    return normalized_mutual_info_score(dim_zero_cat(self.preds), dim_zero_cat(self.target), self.average_method)