Skip to content

Commit 5eeed03

Browse files
committed
fix: with_reverse with AnonymousWalker
1 parent 9b73d8a commit 5eeed03

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

pyrdf2vec/walkers/anonymous.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -50,8 +50,10 @@ def _extract(self, kg: KG, entity: Vertex) -> EntityWalks:
5050
for walk in self.extract_walks(kg, entity):
5151
vertex_names = [vertex.name for vertex in walk]
5252
canonical_walk: List[str] = [
53-
vertex.name if i == 0 else str(vertex_names.index(vertex.name))
54-
for i, vertex in enumerate(walk)
53+
vertex.name
54+
if vertex.name == entity.name
55+
else str(vertex_names.index(vertex.name))
56+
for vertex in walk
5557
]
5658
canonical_walks.add(tuple(canonical_walk))
5759
return {entity.name: list(canonical_walks)}

0 commit comments

Comments
 (0)