update

pyg-team · Jan 14, 2025 · 302e8c6 · 302e8c6
1 parent 1af81ab
commit 302e8c6
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/test/metrics/test_link_pred_metric.py b/test/metrics/test_link_pred_metric.py
@@ -176,6 +176,7 @@ def test_link_pred_metric_collection(num_src_nodes, num_dst_nodes, num_edges):
         metric.update(pred_index_mat[:, :metric.k], edge_label_index)
         out = metric.compute()
         expected[f'{metric.__class__.__name__}@{metric.k}'] = out
+        metric.reset()
 
     metric_collection.update(pred_index_mat, edge_label_index)
     assert metric_collection.compute() == expected