Skip to content

Commit

Permalink
Merge branch 'master' into models/sgformer
Browse files Browse the repository at this point in the history
  • Loading branch information
puririshi98 authored Feb 13, 2025
2 parents b414941 + 02176b7 commit ec1a44f
Showing 1 changed file with 2 additions and 5 deletions.
7 changes: 2 additions & 5 deletions torch_geometric/metrics/link_pred.py
Original file line number Diff line number Diff line change
Expand Up @@ -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."""
Expand Down

0 comments on commit ec1a44f

Please sign in to comment.