diff --git a/torch_geometric/metrics/link_pred.py b/torch_geometric/metrics/link_pred.py index 6ec323fe08c5..599c6a2f36bc 100644 --- a/torch_geometric/metrics/link_pred.py +++ b/torch_geometric/metrics/link_pred.py @@ -402,11 +402,8 @@ def update( # type: ignore for metric in self.values(): if not isinstance(metric, LinkPredMetric): - metric.update( - pred_index_mat=pred_index_mat, - edge_label_index=edge_label_index, - edge_label_weight=edge_label_weight, - ) + metric.update(pred_index_mat, edge_label_index, + edge_label_weight) def compute(self) -> Dict[str, Tensor]: r"""Computes the final metric values."""