Skip to content

Commit 8b1c2f4

Browse files
authored
fix degree output (#315)
1 parent 1464455 commit 8b1c2f4

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/pathpyG/core/graph.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -513,7 +513,7 @@ def degrees(
513513
if return_tensor:
514514
return d
515515
else:
516-
return {str(self.mapping.to_id(i)): d[i].item() for i in range(self.n)}
516+
return {node: degree.item() for node, degree in zip(self.nodes, d)}
517517

518518
def transition_probabilities(self, edge_attr: str | None = None) -> torch.Tensor:
519519
"""Compute transition probabilities based on (weighted) outdegrees.

0 commit comments

Comments
 (0)