Missing output comment.

topN.py

@@ -18,6 +18,8 @@ It filters the predictions so just new predictions are considered.
 Input:
     original: Graph.
     pred: Edge predictions.
+Output:
+    List of the new predictions.
 """
 def filterPreds(original, pred):
     headsO = original.edge_index[edge][0, :].long()  # Heads of the original edges of the graph.