Skip to content

Commit 9b73d8a

Browse files
committed
fix: WalkletWalker with with_reverse
1 parent af404ff commit 9b73d8a

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pyrdf2vec/walkers/walklet.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ def _extract(self, kg: KG, entity: Vertex) -> EntityWalks:
5252
canonical_walks.add((walk[0].name,))
5353
for i in range(1, len(walk)):
5454
if self.with_reverse:
55-
canonical_walks.add((walk[i].name, walk[0].name))
55+
canonical_walks.add((walk[i].name, entity.name))
5656
else:
57-
canonical_walks.add((walk[0].name, walk[i].name))
57+
canonical_walks.add((entity.name, walk[i].name))
5858
return {entity.name: list(canonical_walks)}

0 commit comments

Comments
 (0)